tweaks

evaluate.py

@@ -131,7 +131,7 @@ def get_score(item, image_paths, video_path, metrics, train_steps=100, inference
                 gt_frames.append(img)
     else:
         gt_frames = extract_frames(video_path, fps)
-        gt_frames = gt_frames[:200]
+        gt_frames = gt_frames[:500]
         for f in gt_frames:
             f.thumbnail((512,512))
 
@@ -168,7 +168,7 @@ def get_score(item, image_paths, video_path, metrics, train_steps=100, inference
         psnr2.append(float(compute_psnr(gt, base)))
         lpips2.append(float(compute_lpips(gt, base)))
         
-        if c<50:
+        if c<100:
             print(c)
             fid.append(float(compute_fid(gt, result)))
             fid2.append(float(compute_fid(gt, base)))
@@ -190,7 +190,7 @@ def get_score(item, image_paths, video_path, metrics, train_steps=100, inference
     print("FID:", sum(fid2)/len(fid2))
     print("FVD:", fvd2)
     
-    metrics[item] = {'ft': {}, 'base': {}, 'frames': len(gt_frames), 'complexity': len(images)}
+    metrics[item] = {'ft': {}, 'base': {}, 'n_frames': len(gt_frames), 'complexity': len(images)}
     metrics[item]['ft']['ssim'] = {'avg': sum(ssim)/len(ssim), 'vals': ssim}
     metrics[item]['ft']['psnr'] = {'avg': sum(psnr)/len(psnr), 'vals': psnr}
     metrics[item]['ft']['lpips'] = {'avg': sum(lpips)/len(lpips), 'vals': lpips}
@@ -238,20 +238,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 Exception as e:
-        #    print("Error", item, e)
+        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 Exception as e:
+            print("Error", item, e)
 
 
     ssim = []

main.py

@@ -85,7 +85,7 @@ debug = False
 save_model = True
 should_gen_vid = False
 max_batch_size = 8
-max_frame_count = 200
+max_frame_count = 500
 
 def save_temp_imgs(imgs):
     os.makedirs('temp', exist_ok=True)