H-Liu1997 commited on
Commit
bd16f91
·
1 Parent(s): bb519b0

Replace multiprocessing.Pool with sequential execution for ZeroGPU compatibility

Browse files
Files changed (1) hide show
  1. emage_utils/fast_render.py +22 -28
emage_utils/fast_render.py CHANGED
@@ -177,20 +177,17 @@ def distribute_frames_no_gt(frames, vertices_all):
177
 
178
  def generate_silent_videos(frames, vertices_all, vertices1_all, faces, output_dir):
179
  ids, verts = distribute_frames(frames, vertices_all, vertices1_all)
180
- with multiprocessing.Pool(args['render_concurrent_num']) as pool:
181
- pool.starmap(sub_process_process_frame, [
182
- (
183
- i,
184
- args['render_video_width'],
185
- args['render_video_height'],
186
- args['render_tmp_img_filetype'],
187
- ids[i],
188
- verts[i],
189
- faces,
190
- output_dir
191
- )
192
- for i in range(args['render_concurrent_num'])
193
- ])
194
  out_file = os.path.join(output_dir, "silence_video.mp4")
195
  convert_img_to_mp4(os.path.join(output_dir, f"frame_%d.{args['render_tmp_img_filetype']}"), out_file, args['render_video_fps'])
196
  for fn in glob.glob(os.path.join(output_dir, f"*.{args['render_tmp_img_filetype']}")):
@@ -199,20 +196,17 @@ def generate_silent_videos(frames, vertices_all, vertices1_all, faces, output_di
199
 
200
  def generate_silent_videos_no_gt(frames, vertices_all, faces, output_dir):
201
  ids, verts = distribute_frames_no_gt(frames, vertices_all)
202
- with multiprocessing.Pool(args['render_concurrent_num']) as pool:
203
- pool.starmap(sub_process_process_frame_no_gt, [
204
- (
205
- i,
206
- args['render_video_width'],
207
- args['render_video_height'],
208
- args['render_tmp_img_filetype'],
209
- ids[i],
210
- verts[i],
211
- faces,
212
- output_dir
213
- )
214
- for i in range(args['render_concurrent_num'])
215
- ])
216
  out_file = os.path.join(output_dir, "silence_video.mp4")
217
  convert_img_to_mp4(os.path.join(output_dir, f"frame_%d.{args['render_tmp_img_filetype']}"), out_file, args['render_video_fps'])
218
  for fn in glob.glob(os.path.join(output_dir, f"*.{args['render_tmp_img_filetype']}")):
 
177
 
178
  def generate_silent_videos(frames, vertices_all, vertices1_all, faces, output_dir):
179
  ids, verts = distribute_frames(frames, vertices_all, vertices1_all)
180
+ for i in range(args['render_concurrent_num']):
181
+ sub_process_process_frame(
182
+ i,
183
+ args['render_video_width'],
184
+ args['render_video_height'],
185
+ args['render_tmp_img_filetype'],
186
+ ids[i],
187
+ verts[i],
188
+ faces,
189
+ output_dir
190
+ )
 
 
 
191
  out_file = os.path.join(output_dir, "silence_video.mp4")
192
  convert_img_to_mp4(os.path.join(output_dir, f"frame_%d.{args['render_tmp_img_filetype']}"), out_file, args['render_video_fps'])
193
  for fn in glob.glob(os.path.join(output_dir, f"*.{args['render_tmp_img_filetype']}")):
 
196
 
197
  def generate_silent_videos_no_gt(frames, vertices_all, faces, output_dir):
198
  ids, verts = distribute_frames_no_gt(frames, vertices_all)
199
+ for i in range(args['render_concurrent_num']):
200
+ sub_process_process_frame_no_gt(
201
+ i,
202
+ args['render_video_width'],
203
+ args['render_video_height'],
204
+ args['render_tmp_img_filetype'],
205
+ ids[i],
206
+ verts[i],
207
+ faces,
208
+ output_dir
209
+ )
 
 
 
210
  out_file = os.path.join(output_dir, "silence_video.mp4")
211
  convert_img_to_mp4(os.path.join(output_dir, f"frame_%d.{args['render_tmp_img_filetype']}"), out_file, args['render_video_fps'])
212
  for fn in glob.glob(os.path.join(output_dir, f"*.{args['render_tmp_img_filetype']}")):