【caffe】fcn+VOC2012 将自己的单通道数据集转换为三通道(RGB)图形并附色

经历了一周caffe的苦,楼主终于配置好了,利用fcn训练自己的数据时,发现自己的数据居然是单通道的数据,不能识别,这可急坏了楼主,赶紧百度,终于发现了大佬的解决办法,写在这里供大家参考。
这里用的是matlab代码对图像进行批处理,非常快捷,首先第一步,将单通道图片转换为三通道图片,代码如下:(非原创,如有侵权删)

pathdir = 'F:/SegmentationClass/';%文件路径
format = 'png';%文件格式
files = dir(strcat(pathdir,'*.',format));
% dir 列出当前文件夹中的文件和文件夹。
% strcat 水平串联字符串
% 列出文件夹中后缀为.png的文件
steps = 100;
hwait = waitbar(0,'准备开始');
for n = 1:numel(files)
    filename = strcat(pathdir,files(n).name);
    img = imread(filename);
    x = repmat(img,[1,1,3]);%将单通道图片转换为三通道图片
    imwrite(x,filename,'png');
    str = '正在运行中....';
    waitbar(n/numel(files),hwait,str)
end
close(hwait);

处理完后,接着给图片上色,对应的图片为下,楼主这里找到的代码是赋值成飞机的颜色,你们也可以更改。
在这里插入图片描述
代码如下:

pathdir = 'F:/SegmentationClass/';%文件路径
format = 'png';%文件格式
files = dir(strcat(pathdir,'*.',format));
% dir 列出当前文件夹中的文件和文件夹。
% strcat 水平串联字符串
%    列出文件夹中后缀为.png的文件
steps = 100;
hwait = waitbar(0,'准备开始');
for n = 1:numel(files)
    imgname = strcat(pathdir,files(n).name);
    I = imread(imgname);
    I_gray=rgb2gray(I);
    I_bw=uint8(im2bw(I_gray))*128;
    I1=uint8(zeros(size(I,1),size(I,2),3));
    I1(:,:,1)=I_bw;
    [x,map]=rgb2ind(I1,256);
    imgSaveName=imgname(1:length(imgname)-4);
    imwrite(x,map,strcat(imgSaveName,'.png'));
    str = '正在运行中....';
    waitbar(n/numel(files),hwait,str)
end
close(hwait);

matlab运行两步,完美解决单通道问题,并且直接可以用来训练了,妈妈再也不用担心我的数据!
不得不说,深度学习这玩意真不是人学的。
在这里插入图片描述

  • 1
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

木灬唯紫

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值