From 834d98b6b20bf45fdde9ea042aaa5d016472617c Mon Sep 17 00:00:00 2001 From: gengdongjie Date: Mon, 24 Aug 2020 11:00:19 +0800 Subject: [PATCH] maskrcnn adapt SmoothL1Loss change --- model_zoo/official/cv/maskrcnn/src/MaskRcnn/rcnn_cls.py | 2 +- model_zoo/official/cv/maskrcnn/src/MaskRcnn/rpn.py | 2 +- model_zoo/official/cv/maskrcnn/src/config.py | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/model_zoo/official/cv/maskrcnn/src/MaskRcnn/rcnn_cls.py b/model_zoo/official/cv/maskrcnn/src/MaskRcnn/rcnn_cls.py index ac38b7633..e529f2e22 100644 --- a/model_zoo/official/cv/maskrcnn/src/MaskRcnn/rcnn_cls.py +++ b/model_zoo/official/cv/maskrcnn/src/MaskRcnn/rcnn_cls.py @@ -113,7 +113,7 @@ class RcnnCls(nn.Cell): self.relu = P.ReLU() self.logicaland = P.LogicalAnd() self.loss_cls = P.SoftmaxCrossEntropyWithLogits() - self.loss_bbox = P.SmoothL1Loss(sigma=1.0) + self.loss_bbox = P.SmoothL1Loss(beta=1.0) self.loss_mask = P.SigmoidCrossEntropyWithLogits() self.reshape = P.Reshape() self.onehot = P.OneHot() diff --git a/model_zoo/official/cv/maskrcnn/src/MaskRcnn/rpn.py b/model_zoo/official/cv/maskrcnn/src/MaskRcnn/rpn.py index e06f8eb0d..bd86543c5 100644 --- a/model_zoo/official/cv/maskrcnn/src/MaskRcnn/rpn.py +++ b/model_zoo/official/cv/maskrcnn/src/MaskRcnn/rpn.py @@ -137,7 +137,7 @@ class RPN(nn.Cell): self.CheckValid = P.CheckValid() self.sum_loss = P.ReduceSum() self.loss_cls = P.SigmoidCrossEntropyWithLogits() - self.loss_bbox = P.SmoothL1Loss(sigma=1.0/9.0) + self.loss_bbox = P.SmoothL1Loss(beta=1.0/9.0) self.squeeze = P.Squeeze() self.cast = P.Cast() self.tile = P.Tile() diff --git a/model_zoo/official/cv/maskrcnn/src/config.py b/model_zoo/official/cv/maskrcnn/src/config.py index 97028c782..e1bf13c89 100644 --- a/model_zoo/official/cv/maskrcnn/src/config.py +++ b/model_zoo/official/cv/maskrcnn/src/config.py @@ -20,7 +20,7 @@ from easydict import EasyDict as ed config = ed({ "img_width": 1280, "img_height": 768, - "keep_ratio": False, + "keep_ratio": True, "flip_ratio": 0.5, "photo_ratio": 0.5, "expand_ratio": 1.0, @@ -138,7 +138,7 @@ config = ed({ "epoch_size": 12, "save_checkpoint": True, "save_checkpoint_epochs": 1, - "keep_checkpoint_max": 10, + "keep_checkpoint_max": 12, "save_checkpoint_path": "./checkpoint", "mindrecord_dir": "/home/mxw/mask_rcnn/scripts/MindRecord_COCO2017_Train", -- GitLab