rbler commited on
Commit
febbbfc
·
verified ·
1 Parent(s): afd1369

Update evaluate.py

Browse files
Files changed (1) hide show
  1. evaluate.py +4 -1
evaluate.py CHANGED
@@ -1,10 +1,13 @@
1
  import numpy as np
2
  import json
3
  from vg_evaluator import evaluation_for_challenge
 
 
4
 
 
5
  def run_evaluation(pred_path):
6
  pred_ = json.load(open(pred_path))
7
- gt_ = json.load(open('test_annotations_mmscan.json'))
8
  results = evaluation_for_challenge(gt_,pred_)
9
 
10
 
 
1
  import numpy as np
2
  import json
3
  from vg_evaluator import evaluation_for_challenge
4
+ from datasets import load_dataset
5
+ kk = load_dataset(os.getenv('VG_DATA'))
6
 
7
+ print(len(kk))
8
  def run_evaluation(pred_path):
9
  pred_ = json.load(open(pred_path))
10
+ gt_ = kk
11
  results = evaluation_for_challenge(gt_,pred_)
12
 
13