def evaluate(self): try: cocoDt = self.cocoGt.loadRes(self.coco_results) except: print("no boxes detected, coco eval aborted") return 1 cocoEval = COCOeval(self.cocoGt, cocoDt, "bbox") cocoEval.params.imgIds = list(self.coco_imgIds) cocoEval.evaluate() cocoEval.accumulate() cocoEval.summarize() return cocoEval.stats
def _do_python_eval(self, res_file, res_folder, ann_type): print 'do python eval' coco_dt = self.coco.loadRes(res_file) #print('>>>> do python eval resfile {}'.format(res_file)) coco_eval = COCOeval(self.coco, coco_dt) coco_eval.params.useSegm = (ann_type == 'segm') coco_eval.evaluate() coco_eval.accumulate() info_str = self._print_detection_metrics(coco_eval) eval_file = os.path.join(res_folder, 'detections_%s_results.pkl' % self.image_set) with open(eval_file, 'w') as f: cPickle.dump(coco_eval, f, cPickle.HIGHEST_PROTOCOL) print 'coco eval results saved to %s' % eval_file info_str += 'coco eval results saved to %s\n' % eval_file return info_str
def main(): print('>>>> cocoeval') COCOeval()
def main(): print('>>>> cocoeval') COCOeval().evaluate() COCOeval().accumulate() COCOeval().summarize()