FCN模型实现 Pytorch+预训练VGG16
FCN模型的网络与VGG16类似,之后后边将全连接层换成了卷基层,具体的网络结构与细节可以去看论文: https://people.eecs.berkeley.edu/~jonlong/long_shelhamer_fcn.pdf 下边详细讲一下用Pytorch对FCN的实现: 本文参考了https://zhuanlan.zhihu.com/p/32506912 但是修改了部分代码,加上了很多新的注释,并将代码更新到Pytorch1.x 首先是读取图像 #使用的VOC数据目录 voc_root = '/media/cyq/CU/Ubuntu system files/VOCdevkit/VO
用户评论