Spaces:
Runtime error
Runtime error
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
|