def evaluate_network(self, ckpt):
path = config.EVAL_DIR + '/Data/'
self.filename = path + 'coco_%s_%s_%i.json' % (self.loader.split, args.run_name, ckpt)
detections = []
filenames = self.loader.get_filenames()
bar = progressbar.ProgressBar()
for i in bar(range(len(filenames))):
img_id = filenames[i]
detections.extend(self.process_image(img_id, i))
with open(self.filename, 'w') as f:
json.dump(detections, f)
if args.segment:
iou = self.compute_mean_iou()
cocoEval = self.compute_ap()
return self.compact_results(cocoEval.stats, ckpt)
评论列表
文章目录