欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

训练SSD时出现的数据问题

程序员文章站 2024-03-16 23:13:58
...

使用COCO2014训练SSD网络。

参照https://github.com/weiliu89/coco.git的方法处理数据集。

生成lmdb格式数据。

开始训练之后报这种错误。

OpenCV Error: Assertion failed ((scn == 3 || scn == 4) && (depth == CV_8U || depth == CV_32F)) in cvtColor

各种搜索之后发发现是因为SSD只支持3通道的RGB图像。(也有人说是因为训练图像可能最小边<300,这样resize到固定300x300的时候就会有问题)

解决方案有两个(反正这两个方案对我的问题都有效,不过我选择了第二个):

1. 修改resize_mode为FIT_SMALL_SIZE,batch_size=1.

虽然能解决问题,但是batch_size设为1可能会导致无法收敛等问题,而且所有图像尺寸也不是统一的300x300了,训练的结果就会和原作者有误差,所以我没选择次方案。

2.从训练集中选出灰度图,然后删掉,重新生成lmdb文件重新训练。筛选代码如下,我是用matlab写的:

fid=fopen('/your_root/new_train.txt','wt');%新建一个txt列表  
lst = ['/your_root/train.txt'];           %要读取的列表所在的路径
imgpath = '/image_root/MS-COCO2014/';
count = 0;
fpn = fopen (lst, 'rt');             
while feof(fpn) ~= 1                  
      elem = fgetl(fpn);             
      res = strsplit(elem);   %空格分开
      %disp(strcat(imgpath,char(res(1))));
      img = imread(strcat(imgpath,char(res(1))));%获取图像路径
      if ndims(img)== 2 %灰度图
		disp(strcat(imgpath,char(res(1))));
        count=count+1;
      else
          fprintf(fid,'%s\n',elem);%删除灰度图的list
          
      end 
 end  
 fclose(fid);
 disp(count);

发现train里面有227张灰度图,相对于118287的训练集来说,删除227也没关系。然后就可以不用修改网络参数直接训练了。

不过我发现minival里面也有10张灰度图,但是加载minival_lmdb文件就没出错。这是为什么?难道验证集不需要3通道RGB也可以?

—————————————————————————分割线———————————————————————

果然测试加载数据的时候还是出问题了。

BUG:

mean_values_.size() == 1 || mean_values_.size() ==img_channels Specify either 1 mean_value or as many as channels: 1 

因为minival里面只有10张灰度图,所以我直接将这十张灰度图转为了三通道图,直接s使用opencv中的 cvtcolor()函数。

然后......因为修改了图像信息。

又开始了漫长的处理数据过程(心好累!)

如果还出问题。。。。再说吧