acmyu commited on
Commit
ee217d2
·
1 Parent(s): 7c0be64
Files changed (1) hide show
  1. 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_'):