FCN模型实现-Pytorch+预训练VGG16_fcn-16s

data

label

47 浏览量更新于2023-05-03 评论 1 收藏 284KB PDF 举报

身份认证购VIP最低享 7 折!

领优惠券(最高得80元）

资源详情

资源评论

资源推荐

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/VOC2012'

#此函数用来读取图像和标签的名字

def read_images(root=voc_root, train=True):

txt_fname = root + '/ImageSets/Segmentation/' + ('train.txt' if train else 'val.txt')

with open(txt_fname, 'r') as f:

images = f.read().split()

data = [os.path.join(root, 'JPEGImages', i+'.jpg') for i in images] label = [os.path.join(root, 'SegmentationClass', i+'.png') for i

in images] return data, label

对输入图像做出裁剪，使图像大小一致，方便训练对输入图像做出裁剪，使图像大小一致，方便训练

#这里对图像和标签都截取对应的部分

def rand_crop(data, label, height, width):

'''

data is PIL.Image object

label is PIL.Image object

'''

x = random.uniform(0,data.size[0]-width)

x = int(x)

y = random.uniform(0,data.size[1]-height)

y = int(y)

box = (x,y,x+width,y+height)

data = data.crop(box)

label = label.crop(box)

return data, label

label图像与标签的映射图像与标签的映射

#21个类

classes = ['background','aeroplane','bicycle','bird','boat',

'bottle','bus','car','cat','chair','cow','diningtable',

'dog','horse','motorbike','person','potted plant',

'sheep','sofa','train','tv/monitor']

# 每个类对应的RGB值

colormap = [[0,0,0],[128,0,0],[0,128,0], [128,128,0], [0,0,128],

[128,0,128],[0,128,128],[128,128,128],[64,0,0],[192,0,0],

[64,128,0],[192,128,0],[64,0,128],[192,0,128],

[64,128,128],[192,128,128],[0,64,0],[128,64,0],

[0,192,0],[128,192,0],[0,64,128]]

#下边就是将label中每种颜色映射成0-20的数字

cm2lbl = np.zeros(256**3) # 每个像素点有 0 ~ 255 的选择，RGB 三个通道

for i,cm in enumerate(colormap):

cm2lbl[(cm[0]*256+cm[1])*256+cm[2]] = i # 建立索引

def image2label(im):

data = np.array(im, dtype='int32')

idx = (data[:, :, 0] * 256 + data[:, :, 1]) * 256 + data[:, :, 2] return np.array(cm2lbl[idx], dtype='int64') # 根据索引得到 label 矩阵

数据集制作数据集制作

def img_transforms(im, label, crop_size):

im, label = rand_crop(im, label, *crop_size)

im_tfs = tfs.Compose([

tfs.ToTensor(),

tfs.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])

本内容试读结束，登录后可阅读更多

下载后可阅读完整内容，剩余5页未读，立即下载

weixin_38725625

粉丝: 3
资源: 919

会员权益专享

FCN模型实现-Pytorch+预训练VGG16

评论0

会员权益专享

最新资源

FCN模型实现-Pytorch+预训练VGG16

评论0

语义分割FCN经典网络结构代码（Pytorch框架所写）

VGG16层经典源代码

FCN全卷积网络源代码

pytorch 用vgg16预训练实现fcn

pytorch-fcn-easiest-demo-master_pytorch_fcn_demo_

fcn语义分割pytorch实现

fcn-8s、fcn-16s、fcn-32s

lstm-fcn pytorch

预训练fcn模型提取特征

FCN pytorch

fcn pytorch训练自己的数据集

fcn-500-rtu调试

fcn源码复现pytorch文件

fcn里的--init--函数

fcn8s pytorch

帮我用FCN模型实现图像语义识别全过程，包含数据集划分、模型训练、预测模型、测试新数据等，用pytorch实现代码，谢谢！

写一个FCN网络，用pytorch

帮我用FCN模型实现图像语义识别全过程，我现有图像和mask图像分别存放两个文件夹，希望实现数据集划分、模型训练、预测模型、测试新数据等，用pytorch实现代码，谢谢！

R-FCN网络 matlab

会员权益专享

最新资源