Spaces:
Running
on
Zero
Running
on
Zero
Replace multiprocessing.Pool with sequential execution for ZeroGPU compatibility
Browse files- 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 |
-
|
| 181 |
-
|
| 182 |
-
|
| 183 |
-
|
| 184 |
-
|
| 185 |
-
|
| 186 |
-
|
| 187 |
-
|
| 188 |
-
|
| 189 |
-
|
| 190 |
-
|
| 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 |
-
|
| 203 |
-
|
| 204 |
-
|
| 205 |
-
|
| 206 |
-
|
| 207 |
-
|
| 208 |
-
|
| 209 |
-
|
| 210 |
-
|
| 211 |
-
|
| 212 |
-
|
| 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']}")):
|