Spaces:
Running
on
Zero
Running
on
Zero
make sure tmp video path is exist
Browse files- .gitignore +3 -1
- app.py +11 -0
.gitignore
CHANGED
@@ -1 +1,3 @@
|
|
1 |
-
*__pycache__
|
|
|
|
|
|
1 |
+
*__pycache__
|
2 |
+
gradio_cached_examples/
|
3 |
+
tmp/
|
app.py
CHANGED
@@ -170,11 +170,22 @@ def image_to_3d(
|
|
170 |
},
|
171 |
mode=multiimage_algo,
|
172 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
173 |
video = render_utils.render_video(outputs['gaussian'][0], num_frames=120)['color']
|
174 |
video_geo = render_utils.render_video(outputs['mesh'][0], num_frames=120)['normal']
|
175 |
video = [np.concatenate([video[i], video_geo[i]], axis=1) for i in range(len(video))]
|
|
|
176 |
video_path = os.path.join(user_dir, 'sample.mp4')
|
|
|
177 |
imageio.mimsave(video_path, video, fps=15)
|
|
|
178 |
state = pack_state(outputs['gaussian'][0], outputs['mesh'][0])
|
179 |
torch.cuda.empty_cache()
|
180 |
return state, video_path
|
|
|
170 |
},
|
171 |
mode=multiimage_algo,
|
172 |
)
|
173 |
+
# video = render_utils.render_video(outputs['gaussian'][0], num_frames=120)['color']
|
174 |
+
# video_geo = render_utils.render_video(outputs['mesh'][0], num_frames=120)['normal']
|
175 |
+
# video = [np.concatenate([video[i], video_geo[i]], axis=1) for i in range(len(video))]
|
176 |
+
# video_path = os.path.join(user_dir, 'sample.mp4')
|
177 |
+
# imageio.mimsave(video_path, video, fps=15)
|
178 |
+
# state = pack_state(outputs['gaussian'][0], outputs['mesh'][0])
|
179 |
+
# torch.cuda.empty_cache()
|
180 |
+
# return state, video_path
|
181 |
video = render_utils.render_video(outputs['gaussian'][0], num_frames=120)['color']
|
182 |
video_geo = render_utils.render_video(outputs['mesh'][0], num_frames=120)['normal']
|
183 |
video = [np.concatenate([video[i], video_geo[i]], axis=1) for i in range(len(video))]
|
184 |
+
|
185 |
video_path = os.path.join(user_dir, 'sample.mp4')
|
186 |
+
os.makedirs(os.path.dirname(video_path), exist_ok=True)
|
187 |
imageio.mimsave(video_path, video, fps=15)
|
188 |
+
|
189 |
state = pack_state(outputs['gaussian'][0], outputs['mesh'][0])
|
190 |
torch.cuda.empty_cache()
|
191 |
return state, video_path
|