ritwikraha
commited on
Commit
·
c721853
1
Parent(s):
d22ade1
check
Browse files
app.py
CHANGED
@@ -241,24 +241,24 @@ def show_rendered_image(r,theta,phi):
|
|
241 |
rgb, depth = render_rgb_depth(
|
242 |
nerf_loaded, rays_flat[None, ...], t_vals[None, ...], rand=False, train=False
|
243 |
)
|
244 |
-
return(rgb[0], depth[0])
|
245 |
|
246 |
# app.py text matter starts here
|
247 |
st.title('NeRF:Neural Radiance Fields')
|
248 |
# set the values of r theta phi
|
249 |
r = -30.0
|
250 |
-
theta = st.slider('Enter a value for theta', 0.0, 360.0
|
251 |
st.write('Theta:', theta)
|
252 |
-
phi = st.slider('Enter a value for phi', 0.0, 360.0
|
253 |
st.write('Phi:', phi)
|
254 |
|
255 |
color,depth = show_rendered_image(r,theta,phi)
|
256 |
|
257 |
-
fig, ax = plt.subplots()
|
258 |
-
im = ax.imshow(color)
|
259 |
-
st.pyplot(fig)
|
260 |
|
261 |
-
|
262 |
# st.image(depth, caption = "Depth")
|
263 |
|
264 |
|
|
|
241 |
rgb, depth = render_rgb_depth(
|
242 |
nerf_loaded, rays_flat[None, ...], t_vals[None, ...], rand=False, train=False
|
243 |
)
|
244 |
+
return(rgb[0].numpy()*255.0, depth[0].numpy()*255.0)
|
245 |
|
246 |
# app.py text matter starts here
|
247 |
st.title('NeRF:Neural Radiance Fields')
|
248 |
# set the values of r theta phi
|
249 |
r = -30.0
|
250 |
+
theta = st.slider('Enter a value for theta',min_value = 0.0,max_value = 360.0)
|
251 |
st.write('Theta:', theta)
|
252 |
+
phi = st.slider('Enter a value for phi',min_value = 0.0,max_value = 360.0)
|
253 |
st.write('Phi:', phi)
|
254 |
|
255 |
color,depth = show_rendered_image(r,theta,phi)
|
256 |
|
257 |
+
# fig, ax = plt.subplots()
|
258 |
+
# im = ax.imshow(color)
|
259 |
+
# st.pyplot(fig)
|
260 |
|
261 |
+
st.image(color.astype('int32'), caption = "Color",clamp = True)
|
262 |
# st.image(depth, caption = "Depth")
|
263 |
|
264 |
|