diff --git a/dygraph/configs/solov2/solov2_r50_fpn_1x_coco.yml b/dygraph/configs/solov2/solov2_r50_fpn_1x_coco.yml index f2c15c10b6a36cbb25514b8c01925ab253e0e3df..e5f548d53a80937be526f9c927fa8f6cdb6e7e9c 100644 --- a/dygraph/configs/solov2/solov2_r50_fpn_1x_coco.yml +++ b/dygraph/configs/solov2/solov2_r50_fpn_1x_coco.yml @@ -1,6 +1,6 @@ _BASE_: [ - '../_base_/datasets/coco_instance.yml', - '../_base_/runtime.yml', + '../datasets/coco_instance.yml', + '../runtime.yml', '_base_/solov2_r50_fpn.yml', '_base_/optimizer_1x.yml', '_base_/solov2_reader.yml', diff --git a/dygraph/configs/solov2/solov2_r50_fpn_3x_coco.yml b/dygraph/configs/solov2/solov2_r50_fpn_3x_coco.yml index f0ac682ac8b9728291d4dbeb5ac3d25d44e48e8e..abda7711cc6b907689644552f602e94fb358ec59 100644 --- a/dygraph/configs/solov2/solov2_r50_fpn_3x_coco.yml +++ b/dygraph/configs/solov2/solov2_r50_fpn_3x_coco.yml @@ -1,6 +1,6 @@ _BASE_: [ - '../_base_/datasets/coco_instance.yml', - '../_base_/runtime.yml', + '../datasets/coco_instance.yml', + '../runtime.yml', '_base_/solov2_r50_fpn.yml', '_base_/optimizer_1x.yml', '_base_/solov2_reader.yml', diff --git a/dygraph/ppdet/modeling/layers.py b/dygraph/ppdet/modeling/layers.py index 478d7c14e87d36cb5b23159befdc67f742bf9668..7537a0a0458001059cceb03a5d64d38491f6644f 100644 --- a/dygraph/ppdet/modeling/layers.py +++ b/dygraph/ppdet/modeling/layers.py @@ -900,6 +900,8 @@ class FCOSBox(object): return pred_boxes, pred_scores +@register +@serializable class MaskMatrixNMS(object): """ Matrix NMS for multi-class masks.