From cd6a1e7143f098cfe2dec612357872e51c5113ed Mon Sep 17 00:00:00 2001 From: harimkan Date: Thu, 10 Nov 2022 13:08:33 +0900 Subject: [PATCH] Update evaluation iou_thr of ins-seg --- .../instance-segmentation/efficientnetb2b_maskrcnn/model.py | 1 + .../configs/instance-segmentation/resnet50_maskrcnn/model.py | 1 + .../configs/rotated-detection/efficientnetb2b_maskrcnn/model.py | 2 ++ .../configs/rotated-detection/resnet50_maskrcnn/model.py | 1 + 4 files changed, 5 insertions(+) diff --git a/external/model-preparation-algorithm/configs/instance-segmentation/efficientnetb2b_maskrcnn/model.py b/external/model-preparation-algorithm/configs/instance-segmentation/efficientnetb2b_maskrcnn/model.py index 4bece262367..42a83f70968 100644 --- a/external/model-preparation-algorithm/configs/instance-segmentation/efficientnetb2b_maskrcnn/model.py +++ b/external/model-preparation-algorithm/configs/instance-segmentation/efficientnetb2b_maskrcnn/model.py @@ -2,4 +2,5 @@ "../../../submodule/samples/cfgs/models/backbones/efficientnet_b2b.yaml", "../../../submodule/recipes/stages/_base_/models/detectors/efficientnetb2b_maskrcnn.custom.py", ] +evaluation = dict(interval=1, metric='mAP', save_best='mAP', iou_thr=[0.5]) fp16 = dict(loss_scale=512.0) diff --git a/external/model-preparation-algorithm/configs/instance-segmentation/resnet50_maskrcnn/model.py b/external/model-preparation-algorithm/configs/instance-segmentation/resnet50_maskrcnn/model.py index 0838604070b..534a405a50a 100644 --- a/external/model-preparation-algorithm/configs/instance-segmentation/resnet50_maskrcnn/model.py +++ b/external/model-preparation-algorithm/configs/instance-segmentation/resnet50_maskrcnn/model.py @@ -2,3 +2,4 @@ "../../../submodule/samples/cfgs/models/backbones/resnet50.yaml", "../../../submodule/recipes/stages/_base_/models/detectors/resnet50_maskrcnn.custom.py", ] +evaluation = dict(interval=1, metric='mAP', save_best='mAP', iou_thr=[0.5]) diff --git a/external/model-preparation-algorithm/configs/rotated-detection/efficientnetb2b_maskrcnn/model.py b/external/model-preparation-algorithm/configs/rotated-detection/efficientnetb2b_maskrcnn/model.py index 1c214ce03c0..42a83f70968 100644 --- a/external/model-preparation-algorithm/configs/rotated-detection/efficientnetb2b_maskrcnn/model.py +++ b/external/model-preparation-algorithm/configs/rotated-detection/efficientnetb2b_maskrcnn/model.py @@ -2,3 +2,5 @@ "../../../submodule/samples/cfgs/models/backbones/efficientnet_b2b.yaml", "../../../submodule/recipes/stages/_base_/models/detectors/efficientnetb2b_maskrcnn.custom.py", ] +evaluation = dict(interval=1, metric='mAP', save_best='mAP', iou_thr=[0.5]) +fp16 = dict(loss_scale=512.0) diff --git a/external/model-preparation-algorithm/configs/rotated-detection/resnet50_maskrcnn/model.py b/external/model-preparation-algorithm/configs/rotated-detection/resnet50_maskrcnn/model.py index 0838604070b..534a405a50a 100644 --- a/external/model-preparation-algorithm/configs/rotated-detection/resnet50_maskrcnn/model.py +++ b/external/model-preparation-algorithm/configs/rotated-detection/resnet50_maskrcnn/model.py @@ -2,3 +2,4 @@ "../../../submodule/samples/cfgs/models/backbones/resnet50.yaml", "../../../submodule/recipes/stages/_base_/models/detectors/resnet50_maskrcnn.custom.py", ] +evaluation = dict(interval=1, metric='mAP', save_best='mAP', iou_thr=[0.5])