Spaces:
Paused
Paused
tweaks
Browse files- 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)]
|