Rename clas models

own
Bobholamovic 2 years ago
parent a895234700
commit 9f5c87e9dd
  1. 2
      paddlers/rs_models/clas/__init__.py
  2. 4
      test_tipc/README.md
  3. 2
      test_tipc/configs/clas/mobilenetv3/mobilenetv3_ucmerced.yaml
  4. 2
      tutorials/train/classification/hrnet.py
  5. 3
      tutorials/train/classification/mobilenetv3.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

@ -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 测试工具简介

@ -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

@ -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(

@ -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(

Loading…
Cancel
Save