def run_eval(self, results, save_dir):
self.save_results(results, save_dir)
coco_dets = self.coco.loadRes('{}/results.json'.format(save_dir))
coco_eval = COCOeval(self.coco, coco_dets, "bbox")
# coco_eval.params.catIds = [1]
coco_eval.evaluate()
coco_eval.accumulate()
coco_eval.summarize()
return coco_eval.stats[1]
def run_eval(self, results, save_dir):
self.save_results(results, save_dir)
coco_dets = self.coco.loadRes('{}/results.json'.format(save_dir))
coco_eval = COCOeval(self.coco, coco_dets, "bbox")
# coco_eval.params.catIds = [1]
save_summarize = self.opt.load_model
save_summarize = '/'.join([s for s in save_summarize.split('/')][:-1])
save_summarize = os.path.join(save_summarize, 'result.json')
print('save_summarize : ', save_summarize)
coco_eval.evaluate()
coco_eval.accumulate()
coco_eval.summarize(save_summarize=save_summarize)
return coco_eval.stats[1]
def _do_python_keypoint_eval(self, res_file, res_folder):
coco_dt = self.coco.loadRes(res_file)
coco_eval = COCOeval(self.coco, coco_dt, 'keypoints')
coco_eval.params.useSegm = None
coco_eval.evaluate()
coco_eval.accumulate()
coco_eval.summarize()
stats_names = [
'AP', 'Ap .5', 'AP .75', 'AP (M)', 'AP (L)', 'AR', 'AR .5',
'AR .75', 'AR (M)', 'AR (L)'
]
info_str = []
for ind, name in enumerate(stats_names):
info_str.append((name, coco_eval.stats[ind]))
# info_str.append(coco_eval.stats[ind])
return info_str