acmyu commited on
Commit
67fa8fb
·
1 Parent(s): bb4dd0f
Files changed (1) hide show
  1. evaluate.py +2 -0
evaluate.py CHANGED
@@ -129,6 +129,7 @@ def get_score(item, image_paths, video_path, metrics, train_steps=100, inference
129
  results_base.append(img)
130
  elif filename.startswith('frame_'):
131
  gt_frames.append(img)
 
132
  else:
133
  gt_frames = extract_frames(video_path, fps)
134
  gt_frames = gt_frames[:200]
@@ -227,6 +228,7 @@ def run_evaluate():
227
 
228
  with open('/data/metrics.json', 'r') as file:
229
  metrics = json.load(file)
 
230
 
231
  items = os.listdir('test')
232
  items = [it for it in items if not it[0]=='.' and not os.path.isfile('test/'+it)]
 
129
  results_base.append(img)
130
  elif filename.startswith('frame_'):
131
  gt_frames.append(img)
132
+ gt_frames = gt_frames[:200]
133
  else:
134
  gt_frames = extract_frames(video_path, fps)
135
  gt_frames = gt_frames[:200]
 
228
 
229
  with open('/data/metrics.json', 'r') as file:
230
  metrics = json.load(file)
231
+ #metrics = {'aaa': metrics['aaa']}
232
 
233
  items = os.listdir('test')
234
  items = [it for it in items if not it[0]=='.' and not os.path.isfile('test/'+it)]