acmyu commited on
Commit
a3692c7
·
1 Parent(s): 05511bd

remove output image pad

Browse files
Files changed (1) hide show
  1. main.py +2 -2
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):