From 5453ffbefca9f9ee93961c23771cd9c7fea096f9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9C=B1=E5=AD=90=E8=B0=A6?= Date: Wed, 10 May 2023 19:57:21 +0800 Subject: [PATCH] pr_curve works --- tools/eval.py | 4 ++-- yolov6/core/evaler.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tools/eval.py b/tools/eval.py index 5543029c..671464e5 100644 --- a/tools/eval.py +++ b/tools/eval.py @@ -107,8 +107,8 @@ def run(data, eval_config_file='./configs/experiment/eval_640_repro.py', verbose=False, do_coco_metric=True, - do_pr_metric=False, - plot_curve=False, + do_pr_metric=True, + plot_curve=True plot_confusion_matrix=False, config_file=None, specific_shape=False, diff --git a/yolov6/core/evaler.py b/yolov6/core/evaler.py index a776d314..2464d722 100644 --- a/yolov6/core/evaler.py +++ b/yolov6/core/evaler.py @@ -33,7 +33,7 @@ def __init__(self, infer_on_rect=False, verbose=False, do_coco_metric=True, - do_pr_metric=False, + do_pr_metric=True, plot_curve=True, plot_confusion_matrix=False, specific_shape=False,