【caffe】模型训练中的通道数错误

主要基于 https://blog.csdn.net/10km/article/details/70160645

最近针对性地调整了一下数据集,通过从ILSVSC12中抽取一些类生成txt和lmdb用于检测任务。处理办法有机会另写一篇把。

训练时主要遇到了两类错误:

  1. CHECK(std::equal(top_shape.begin() + 1, top_shape.begin() + 4, shape.begin() + 1)
  2. VScn = cv::Set<3,4>…

有文章说通过把P.Resize.warp改为P.FIT_SMALL_SIZE之后可以修正,但是这样不能设置大于1的batch_size,果断弃用。有文章说这可能是因为图片中包含灰度图像从而产生的通道错误,但是我写脚本检查了所有XML文件,并没有发现任何1的地方,最后我在train_transform_param和test_transform_param中加入了如下一行,之后能成功跑起来了。

'force_color': True,
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值