Spaces:
Paused
Paused
fix bug
Browse files- evaluate.py +3 -3
evaluate.py
CHANGED
|
@@ -120,9 +120,9 @@ def get_score(item, image_paths, video_path, metrics, train_steps=100, inference
|
|
| 120 |
results = []
|
| 121 |
results_base = []
|
| 122 |
gt_frames = []
|
| 123 |
-
if os.path.isdir('out/'+item):
|
| 124 |
-
for filename in os.listdir('out/'+item):
|
| 125 |
-
img = Image.open('out/'+item+'/'+filename)
|
| 126 |
if filename.startswith('result_'):
|
| 127 |
results.append(img)
|
| 128 |
elif filename.startswith('base_'):
|
|
|
|
| 120 |
results = []
|
| 121 |
results_base = []
|
| 122 |
gt_frames = []
|
| 123 |
+
if os.path.isdir('/data/out/'+item):
|
| 124 |
+
for filename in os.listdir('/data/out/'+item):
|
| 125 |
+
img = Image.open('/data/out/'+item+'/'+filename)
|
| 126 |
if filename.startswith('result_'):
|
| 127 |
results.append(img)
|
| 128 |
elif filename.startswith('base_'):
|