fix bug

evaluate.py

@@ -169,20 +169,20 @@ def run_evaluate():
             continue
         print(item)
         
-        #try:
-        files = get_files('test/'+item)
-        images = list(filter(lambda x: not x.endswith('.mp4'), files))
-        images = ['test/'+item+'/'+img for img in images]
-        videos = [x for x in files if x.endswith('.mp4')]
-        print(images, videos)
-
-        if len(videos) == 1:
-            get_score(item, images, 'test/'+item+'/'+videos[0], metrics)
-            #get_score(item, ['test/'+item+'/1.jpg', 'test/'+item+'/2.jpg', 'test/'+item+'/3.jpg'], 'test/'+item+'/v.mp4')
-        else:
-            print('Error: mp4 not found')
-        #except:
-        #    print("Error", item)
+        try:
+            files = get_files('test/'+item)
+            images = list(filter(lambda x: not x.endswith('.mp4'), files))
+            images = ['test/'+item+'/'+img for img in images]
+            videos = [x for x in files if x.endswith('.mp4')]
+            print(images, videos)
+
+            if len(videos) == 1:
+                get_score(item, images, 'test/'+item+'/'+videos[0], metrics)
+                #get_score(item, ['test/'+item+'/1.jpg', 'test/'+item+'/2.jpg', 'test/'+item+'/3.jpg'], 'test/'+item+'/v.mp4')
+            else:
+                print('Error: mp4 not found')
+        except:
+            print("Error", item)
 
 
     ssim = []

main.py

@@ -337,17 +337,17 @@ def prepare_inputs_inference(in_img, in_vid, frames, fps, dwpose, rembg_session,
         target_poses_coords.append(json.dumps(coords))
         progress_bar.update(1)
         
-    for tpose in target_poses:
-        bbox = tpose.getbbox()
-        left, top, right, bottom = bbox
-        max_left = min(max_left, left)
-        max_top = min(max_top, top)
-        max_right = max(max_right, right)
-        max_bottom = max(max_bottom, bottom)
         
     target_poses_cropped = []
     for tpose in target_poses:
         if resize_inputs:
+            bbox = tpose.getbbox()
+            left, top, right, bottom = bbox
+            max_left = min(max_left, left)
+            max_top = min(max_top, top)
+            max_right = max(max_right, right)
+            max_bottom = max(max_bottom, bottom)
+
             tpose = tpose.crop((max_left, max_top, max_right, max_bottom))
             tpose = ImageOps.expand(tpose, border=int(tpose.width*0.2), fill=(0,0,0))