[Fix]: auc_roc in predict (#3446)

* [Fix]: auc_roc in predict * move to merge_test_config
PaddlePaddle · Aug 14, 2023 · 0dd7dc5 · 0dd7dc5
1 parent 17d0894
commit 0dd7dc5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tools/predict.py b/tools/predict.py
@@ -17,10 +17,10 @@
 
 import paddle
 
-from paddleseg.cvlibs import manager, Config, SegBuilder
-from paddleseg.utils import get_sys_env, logger, get_image_list, utils
 from paddleseg.core import predict
+from paddleseg.cvlibs import Config, SegBuilder, manager
 from paddleseg.transforms import Compose
+from paddleseg.utils import get_image_list, get_sys_env, logger, utils
 
 
 def parse_args():
@@ -105,6 +105,8 @@ def merge_test_config(cfg, args):
     test_config = cfg.test_config
     if 'aug_eval' in test_config:
         test_config.pop('aug_eval')
+    if 'auc_roc' in test_config:
+        test_config.pop('auc_roc')
     if args.aug_pred:
         test_config['aug_pred'] = args.aug_pred
         test_config['scales'] = args.scales