Create valcoco.py
parent
95587b46e6
commit
b35f8dd3c2
|
@ -0,0 +1,16 @@
|
|||
import sys
|
||||
from pycocotools.coco import COCO
|
||||
from pycocotools.cocoeval import COCOeval
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
pred = sys.argv[1]
|
||||
gt = 'instances_val2017.json'
|
||||
|
||||
cocoGt = COCO(gt)
|
||||
cocoDt = cocoGt.loadRes(pred)
|
||||
|
||||
cocoEval = COCOeval(cocoGt, cocoDt, 'bbox')
|
||||
cocoEval.evaluate()
|
||||
cocoEval.accumulate()
|
||||
cocoEval.summarize()
|
Loading…
Reference in New Issue