用SSD训练自己的数据时遇到了这个错误,查看cpp文件,报错原因是mean.size()和输入图像的通道数不一致。
导致错误的原因可能如下:
1.数据集存在单通道图像:
在ssd_pascal.py的train_transform_param中参数定义可知道ssd默认图像为三通道RGB.
解决:
找出到单通道图像转为三通道RGB
2.默认读取灰度图像(这种可能性大一些):
遇到错误时候感觉很奇怪,我的数据都是RGB图像,不知道为什么读取的是单通道图像
解决:
在ssd_pascal.py的train_transform_param中参数定义中加上
“force_color”: True