Steven18 commited on
Commit
4197679
·
1 Parent(s): 2460baf

make sure tmp video path is exist

Browse files
Files changed (2) hide show
  1. .gitignore +3 -1
  2. 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