计算机视觉入门到实战系列(二十二)语义分割之FCN-8s算法
语义分割之FCN-8s算法
我们选择ResNet101作为模型的主干网络.之前我们已经实现了ResNet34,这里我们不从零实现ResNet101了,而是直接从模型库里面加载。这里需要注意一般我们考虑两个方面通道数和分辨率,从这两个绝度去思考下面的转化。通道数和分辨率是两码事不要混淆。
加载ResNet101
from torchvision import models # 使用 ResNet101 作为主干网络,模型使用ImageNet预训练 pretrained_net = models.resnet101(pretrained='imagenet')