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' ] )
1