fix bugs

evaluate.py

@@ -120,6 +120,7 @@ def get_score(item, image_paths, video_path, metrics, train_steps=100, inference
     results = []
     results_base = []
     gt_frames = []
+    max_frame_count = 200
     if os.path.isdir('/data/out/'+item):
         for filename in os.listdir('/data/out/'+item):
             img = Image.open('/data/out/'+item+'/'+filename)
@@ -129,10 +130,14 @@ def get_score(item, image_paths, video_path, metrics, train_steps=100, inference
                 results_base.append(img)
             elif filename.startswith('frame_'):
                 gt_frames.append(img)
-        gt_frames = gt_frames[:200]
+        
+        results = gt_frames[:max_frame_count]
+        results_base = gt_frames[:max_frame_count]
+        gt_frames = gt_frames[:max_frame_count]
+        
     else:
         gt_frames = extract_frames(video_path, fps)
-        gt_frames = gt_frames[:200]
+        gt_frames = gt_frames[:max_frame_count]
         for f in gt_frames:
             f.thumbnail((512,512))
 
@@ -228,7 +233,7 @@ def run_evaluate():
 
     with open('/data/metrics.json', 'r') as file:
         metrics = json.load(file)
-        #metrics = {'aaa': metrics['aaa']}
+        metrics = {'aaa': metrics['aaa']}
 
     items = os.listdir('test') 
     items = [it for it in items if not it[0]=='.' and not os.path.isfile('test/'+it)]