torch从零开始搭建deeplabv3+训练自己的数据集!一、制作自己数据集二、训练网络
制作自己数据集torch数据加载原理torch数据输入需要转换为张量,因此需要将读取的图片数据和标签转换为tensor,重写自己的读取数据类,只需要提供图片和标签的文件夹路径,即可实现数据的读取,但是数据读取完毕后因为计算能力有限,需要使用torch框架提供的Data.DataLoader函数实现分批次输入。输入图片格式,mask标签不用采取one-hot编码输入网络,需要格式为背景值:对应数值0,类别A:对应数值1,类别B:对应数值2,类别C:对应数值3,如下图所示。代码实现