Spaces:
Paused
Paused
don't remove bg for single frame generation
Browse files
main.py
CHANGED
|
@@ -1211,9 +1211,10 @@ def run_inference(images, video_path, frames, train_steps=100, inference_steps=1
|
|
| 1211 |
return run_inference_impl(images, video_path, frames, train_steps, inference_steps, fps, modelId, img_width, img_height, bg_remove, resize_inputs)
|
| 1212 |
|
| 1213 |
|
| 1214 |
-
def generate_frame(images, target_poses, train_steps=100, inference_steps=10, modelId="fine_tuned_pcdms", img_width=1920, img_height=1080,
|
| 1215 |
finetune=True
|
| 1216 |
is_app=True
|
|
|
|
| 1217 |
|
| 1218 |
dwpose, rembg_session, pcdms_model, noise_scheduler, image_encoder_p, image_encoder_g, vae, unet = load_models()
|
| 1219 |
|
|
@@ -1235,7 +1236,7 @@ def generate_frame(images, target_poses, train_steps=100, inference_steps=10, mo
|
|
| 1235 |
#urls = save_temp_imgs(results)
|
| 1236 |
|
| 1237 |
# postprocessing
|
| 1238 |
-
if
|
| 1239 |
results = [removebg(img, rembg_session, True) for img in results]
|
| 1240 |
#results = [img_pad(img, img_width, img_height, True) for img in results]
|
| 1241 |
|
|
|
|
| 1211 |
return run_inference_impl(images, video_path, frames, train_steps, inference_steps, fps, modelId, img_width, img_height, bg_remove, resize_inputs)
|
| 1212 |
|
| 1213 |
|
| 1214 |
+
def generate_frame(images, target_poses, train_steps=100, inference_steps=10, modelId="fine_tuned_pcdms", img_width=1920, img_height=1080, remove_bg=True, resize_inputs=True):
|
| 1215 |
finetune=True
|
| 1216 |
is_app=True
|
| 1217 |
+
bg_remove=False
|
| 1218 |
|
| 1219 |
dwpose, rembg_session, pcdms_model, noise_scheduler, image_encoder_p, image_encoder_g, vae, unet = load_models()
|
| 1220 |
|
|
|
|
| 1236 |
#urls = save_temp_imgs(results)
|
| 1237 |
|
| 1238 |
# postprocessing
|
| 1239 |
+
if remove_bg:
|
| 1240 |
results = [removebg(img, rembg_session, True) for img in results]
|
| 1241 |
#results = [img_pad(img, img_width, img_height, True) for img in results]
|
| 1242 |
|