Spaces:
Paused
Paused
fix bugs
Browse files- 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 =
|
| 135 |
-
results_base =
|
| 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 = {
|
| 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)]
|