fix condition of mask (#1881)

7257a364 · Guanghua Yu · GitHub · c5c91cee · 7257a364 · 7257a364
显示空白变更内容
内联并排

Showing with 4 addition and 2 deletion

tools/eval.py tools/eval.py +2 -1

tools/infer.py tools/infer.py +2 -1

未找到文件。
--- a/tools/eval.py
+++ b/tools/eval.py
@@ -109,7 +109,8 @@ def run(FLAGS, cfg, place):
                                                use_default_label)

    mask_resolution = None
-    if cfg['MaskPostProcess']['mask_resolution'] is not None:
+    if 'Mask' in cfg.architecture and cfg['MaskPostProcess'][
+            'mask_resolution'] is not None:
        mask_resolution = int(cfg['MaskPostProcess']['mask_resolution'])
    infer_res = get_infer_results(
        outs_res,

--- a/tools/infer.py
+++ b/tools/infer.py
@@ -161,7 +161,8 @@ def run(FLAGS, cfg, place):
        im_ids = data[fields.index('im_id')].numpy()

        mask_resolution = None
-        if cfg['MaskPostProcess']['mask_resolution'] is not None:
+        if 'Mask' in cfg.architecture and cfg['MaskPostProcess'][
+                'mask_resolution'] is not None:
            mask_resolution = int(cfg['MaskPostProcess']['mask_resolution'])
        batch_res = get_infer_results(
            [outs],