学习目标
• 掌握语义分割模型的原理和训练过程;
• 掌握语义分割模型的发展脉络;
• 掌握语义分割模型的使用;
语义分割模型分类
FCN 、SegNet、Unet、DeepLab、RefineNet、PSPNet、
GAN 语义分割
最常用的是:segmentation_models_pytorch 库来调用
安装方法:
pip install segmentation-models-pytorch
调用方法:
import segmentation_models_pytorch as smp
model = smp.Unet()
参数:
model = smp.Unet('resnet34', encoder_weights="imagenet",in_channels=1,classes=3, activation='softmax')
各个模型输出结果对比
模型加载中,待补充