Spaces:
Paused
Paused
remove output image pad
Browse files
main.py
CHANGED
|
@@ -1148,11 +1148,11 @@ def run_inference(images, video_path, train_steps=100, inference_steps=10, fps=1
|
|
| 1148 |
|
| 1149 |
# postprocessing
|
| 1150 |
results = [removebg(img, rembg_session, True) for img in results]
|
| 1151 |
-
results = [img_pad(img, img_width, img_height, True) for img in results]
|
| 1152 |
|
| 1153 |
print("Done!")
|
| 1154 |
|
| 1155 |
-
return out_vid+'.webm', results, getThumbnails(results), target_poses_coords, orig_frames
|
| 1156 |
|
| 1157 |
|
| 1158 |
def run_app(images, video_path, train_steps=100, inference_steps=10, fps=12, bg_remove=False, resize_inputs=True):
|
|
|
|
| 1148 |
|
| 1149 |
# postprocessing
|
| 1150 |
results = [removebg(img, rembg_session, True) for img in results]
|
| 1151 |
+
#results = [img_pad(img, img_width, img_height, True) for img in results]
|
| 1152 |
|
| 1153 |
print("Done!")
|
| 1154 |
|
| 1155 |
+
return out_vid+'.webm', results, getThumbnails(results), target_poses_coords, getThumbnails(orig_frames)
|
| 1156 |
|
| 1157 |
|
| 1158 |
def run_app(images, video_path, train_steps=100, inference_steps=10, fps=12, bg_remove=False, resize_inputs=True):
|