Browse Source

Rename clas models

Bobholamovic 2 years ago
parent
commit
9f5c87e9dd

+ 1 - 1
paddlers/rs_models/clas/__init__.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from .condensenet_v2 import CondenseNetV2_a, CondenseNetV2_b, CondenseNetV2_c
+from .condensenetv2 import CondenseNetV2_A, CondenseNetV2_B, CondenseNetV2_C

+ 4 - 0
test_tipc/README.md

@@ -32,6 +32,7 @@
 | 变化检测 | FC-Siam-conc | 支持 | - | - | - |
 | 变化检测 | FC-Siam-diff | 支持 | - | - | - |
 | 变化检测 | ChangeFormer | 支持 | - | - | - |
+| 场景分类 | CondenseNet V2 | 支持 | - | - | - |
 | 场景分类 | HRNet | 支持 | - | - | - |
 | 场景分类 | MobileNetV3 | 支持 | - | - | - |
 | 场景分类 | ResNet50-vd | 支持 | - | - | - |
@@ -43,8 +44,11 @@
 | 目标检测 | PP-YOLO Tiny | 支持 | - | - | - |
 | 目标检测 | PP-YOLOv2 | 支持 | - | - | - |
 | 目标检测 | YOLOv3 | 支持 | - | - | - |
+| 图像分割 | BiSeNet V2 | 支持 | - | - | - |
 | 图像分割 | DeepLab V3+ | 支持 | - | - | - |
 | 图像分割 | FarSeg | 支持 | - | - | - |
+| 图像分割 | Fast-SCNN | 支持 | - | - | - |
+| 图像分割 | HRNet | 支持 | - | - | - |
 | 图像分割 | UNet | 支持 | - | - | - |
 
 ## 3 测试工具简介

+ 1 - 1
test_tipc/configs/clas/mobilenetv3/mobilenetv3_ucmerced.yaml

@@ -5,6 +5,6 @@ _base_: ../_base_/ucmerced.yaml
 save_dir: ./test_tipc/output/clas/mobilenetv3/
 
 model: !Node
-    type: MobileNetV3_small_x1_0
+    type: MobileNetV3
     args:
         num_classes: 21

+ 1 - 1
tutorials/train/classification/hrnet.py

@@ -68,7 +68,7 @@ eval_dataset = pdrs.datasets.ClasDataset(
 # 构建HRNet模型
 # 目前已支持的模型请参考:https://github.com/PaddlePaddle/PaddleRS/blob/develop/docs/intro/model_zoo.md
 # 模型输入参数请参考:https://github.com/PaddlePaddle/PaddleRS/blob/develop/paddlers/tasks/classifier.py
-model = pdrs.tasks.clas.HRNet_W18_C(num_classes=len(train_dataset.labels))
+model = pdrs.tasks.clas.HRNet(num_classes=len(train_dataset.labels))
 
 # 执行模型训练
 model.train(

+ 1 - 2
tutorials/train/classification/mobilenetv3.py

@@ -68,8 +68,7 @@ eval_dataset = pdrs.datasets.ClasDataset(
 # 构建MobileNetV3模型
 # 目前已支持的模型请参考:https://github.com/PaddlePaddle/PaddleRS/blob/develop/docs/intro/model_zoo.md
 # 模型输入参数请参考:https://github.com/PaddlePaddle/PaddleRS/blob/develop/paddlers/tasks/classifier.py
-model = pdrs.tasks.clas.MobileNetV3_small_x1_0(
-    num_classes=len(train_dataset.labels))
+model = pdrs.tasks.clas.MobileNetV3(num_classes=len(train_dataset.labels))
 
 # 执行模型训练
 model.train(