acmyu commited on
Commit
5bde6cc
·
1 Parent(s): 9dde25d

don't remove bg for single frame generation

Browse files
Files changed (1) hide show
  1. main.py +3 -2
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, bg_remove=True, resize_inputs=True):
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 no_bg_final:
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