acmyu commited on
Commit
cd1c851
·
1 Parent(s): ae70ea2
Files changed (1) hide show
  1. evaluate.py +5 -5
evaluate.py CHANGED
@@ -131,10 +131,10 @@ def get_score(item, image_paths, video_path, metrics, train_steps=100, inference
131
  elif filename.startswith('frame_'):
132
  gt_frames.append(img)
133
 
134
- results = gt_frames[:max_frame_count]
135
- results_base = gt_frames[:max_frame_count]
136
- gt_frames = gt_frames[:max_frame_count]
137
-
138
  else:
139
  gt_frames = extract_frames(video_path, fps)
140
  gt_frames = gt_frames[:max_frame_count]
@@ -233,7 +233,7 @@ def run_evaluate():
233
 
234
  with open('/data/metrics.json', 'r') as file:
235
  metrics = json.load(file)
236
- metrics = {'aaa': metrics['aaa']}
237
 
238
  items = os.listdir('test')
239
  items = [it for it in items if not it[0]=='.' and not os.path.isfile('test/'+it)]
 
131
  elif filename.startswith('frame_'):
132
  gt_frames.append(img)
133
 
134
+ #results = results[:max_frame_count]
135
+ #results_base = results_base[:max_frame_count]
136
+ #gt_frames = gt_frames[:max_frame_count]
137
+
138
  else:
139
  gt_frames = extract_frames(video_path, fps)
140
  gt_frames = gt_frames[:max_frame_count]
 
233
 
234
  with open('/data/metrics.json', 'r') as file:
235
  metrics = json.load(file)
236
+ #metrics['artstation'] = {}
237
 
238
  items = os.listdir('test')
239
  items = [it for it in items if not it[0]=='.' and not os.path.isfile('test/'+it)]