COCOeval 간단 사용법

cocoeval 검색해서 퍼온 사진, 정상적으로 수행되면 이런식으로 나온다.

# coco_gt, coco_pred 선언법
# 주석에는 파일명으로 하라고 하는데 코드보면 numpy array로 넣어줘도 사용 가능
coco_gt = COCO("file_path")
coco_pred = coco_gt.loadRes(pred_json_path)

coco_eval = COCOeval(coco_gt, coco_pred, 'bbox')

# 각종 파라미터 설정
coco_eval.params.imgIds = image_ids
# params.catIds 설정하면 class 별 mAP 출력해주는듯

# 수치 뽑는 3종셋트
coco_eval.evaluate()
coco_eval.accumulate()
coco_eval.summarize()

# coco_eval.stats에 접근하면 summarize 시에 할당된 수치를 받아올 수 있다.

reference

https://github.com/cocodataset/cocoapi/blob/master/PythonAPI/pycocotools/coco.py#L305
https://github.com/cocodataset/cocoapi/blob/master/PythonAPI/pycocotools/cocoeval.py