get metrics for user edits

evaluate.py

@@ -116,14 +116,6 @@ def compute_fid(img1, img2):
 
 def get_score(item, image_paths, video_path, train_steps=100, inference_steps=10, fps=12, bg_remove=False):
     images = []
-    if not isinstance(image_paths[0], str):
-        images = image_paths
-    else:
-        for path in image_paths:
-            print(path)
-            img = Image.open(path)
-            images.append([img])
-
     results = []
     results_base = []
     gt_frames = []
@@ -143,6 +135,16 @@ def get_score(item, image_paths, video_path, train_steps=100, inference_steps=10
         #gt_frames = gt_frames[:max_frame_count]
 
     else:
+        
+        if not isinstance(image_paths[0], str):
+            images = image_paths
+        else:
+            for path in image_paths:
+                print(path)
+                img = Image.open(path)
+                images.append([img])
+
+
         gt_frames = extract_frames(video_path, fps)
         gt_frames = gt_frames[:max_frame_count]
         for f in gt_frames:

main.py

@@ -86,6 +86,7 @@ save_model = True
 should_gen_vid = False
 max_batch_size = 8
 max_frame_count = 200
+no_bg_final = True
 
 def save_temp_imgs(imgs):
     os.makedirs('temp', exist_ok=True)
@@ -1196,7 +1197,8 @@ def run_inference_impl(images, video_path, frames, train_steps=100, inference_st
     
     
     # postprocessing
-    results = [removebg(img, rembg_session, True) for img in results]
+    if no_bg_final:
+        results = [removebg(img, rembg_session, True) for img in results]
     #results = [img_pad(img, img_width, img_height, True) for img in results]
     
     print("Done!")
@@ -1233,7 +1235,8 @@ def generate_frame(images, target_poses, train_steps=100, inference_steps=10, mo
     #urls = save_temp_imgs(results)
     
     # postprocessing
-    results = [removebg(img, rembg_session, True) for img in results]
+    if no_bg_final:
+        results = [removebg(img, rembg_session, True) for img in results]
     #results = [img_pad(img, img_width, img_height, True) for img in results]
     
     print("Done!")