[MaskRCNN] set FPN to default; don't eval mask if not used

examples/FasterRCNN/config.py

@@ -80,7 +80,7 @@ _C = config     # short alias to avoid coding
 # mode flags ---------------------
 _C.TRAINER = 'replicated'  # options: 'horovod', 'replicated'
 _C.MODE_MASK = True        # Faster R-CNN or Mask R-CNN
-_C.MODE_FPN = False
+_C.MODE_FPN = True
 
 # dataset -----------------------
 _C.DATA.BASEDIR = '/path/to/your/DATA/DIR'

examples/FasterRCNN/dataset/coco.py

@@ -70,6 +70,8 @@ class COCODetection(DatasetSplit):
         """
         from pycocotools.cocoeval import COCOeval
         ret = {}
+        has_mask = "segmentation" in results[0]  # results will be modified by loadRes
+
         cocoDt = self.coco.loadRes(results)
         cocoEval = COCOeval(self.coco, cocoDt, 'bbox')
         cocoEval.evaluate()
@@ -79,7 +81,7 @@ class COCODetection(DatasetSplit):
         for k in range(6):
             ret['mAP(bbox)/' + fields[k]] = cocoEval.stats[k]
 
-        if len(results) > 0 and 'segmentation' in results[0]:
+        if len(results) > 0 and has_mask:
             cocoEval = COCOeval(self.coco, cocoDt, 'segm')
             cocoEval.evaluate()
             cocoEval.accumulate()