-
DeepLabV3Plus-Pytorch提供了在DeepLabv3、DeepLabv3 +上以及VOC和Cityscapes数据集上预训练的权重
资源介绍
DeepLabv3Plus-Pytorch
DeepLabv3,DeepLabv3 +和VOC和Cityscapes上的预训练权重。
可用架构
使用“ --model ARCH_NAME”指定模型体系结构,并使用“ --output_stride OUTPUT_STRIDE”设置输出步幅。
DeepLabV3
DeepLabV3 +
deeplabv3_resnet50
deeplabv3plus_resnet50
deeplabv3_resnet101
deeplabv3plus_resnet101
deeplabv3_mobilenet
deeplabv3plus_mobilenet
可用型号: ,
加载预训练的模型:
model . load_state_dict ( torch . load ( CKPT_PATH )[ 'model_state' ] )
- 上一篇: *语料库直接解压使用即可
- 下一篇: 频繁模式挖掘.docx