Fix evaluation with no mask (#2577)

PaddleCV/object_detection/ppdet/utils/eval_utils.py

@@ -88,7 +88,7 @@ def eval_run(exe, compile_program, pyreader, keys, values, cls):
     return results
 
 
-def eval_results(results, feed, metric, resolution, output_file=None):
+def eval_results(results, feed, metric, resolution=None, output_file=None):
     """Evaluation for evaluation program results"""
     if metric == 'COCO':
         from ppdet.utils.coco_eval import bbox_eval, mask_eval

PaddleCV/object_detection/tools/eval.py

@@ -99,8 +99,10 @@ def main():
     # 6. Run
     results = eval_run(exe, compile_program, pyreader, keys, values, cls)
     # Evaluation
-    eval_results(results, eval_feed, cfg.metric,
-                 model.mask_head.resolution, FLAGS.output_file)
+    resolution = None
+    if 'mask' in results[0]:
+        resolution = model.mask_head.resolution
+    eval_results(results, eval_feed, cfg.metric, resolution, FLAGS.output_file)
 
 
 if __name__ == '__main__':

PaddleCV/object_detection/tools/train.py

@@ -155,8 +155,11 @@ def main():
                 results = eval_run(exe, eval_compile_program, eval_pyreader,
                                    eval_keys, eval_values, eval_cls)
                 # Evaluation
+                resolution = None
+                if 'mask' in results[0]:
+                    resolution = model.mask_head.resolution
                 eval_results(results, eval_feed, cfg.metric,
-                             model.mask_head.resolution, FLAGS.output_file)
+                             resolution, FLAGS.output_file)
 
     checkpoint.save(exe, train_prog, os.path.join(save_dir, "model_final"))
     train_pyreader.reset()