TypechoJoeTheme

半醉残影

统计
文章目录

[常见错误] caffe跑resnet50 imagenet数据集错误height <= datum_height

2018-09-21
/
0 评论
/
494 阅读
/
正在检测是否收录...
09/21

height <= datum_height

原因

下载了imagenet数据集后,使用数据集中验证集验证resnet50准确率出现如下错误:

height <= datum_height(224 vs 170)

出现这个问题的原因是验证图片数据集(LMDB)的图片的size小于需要裁剪的尺寸。

下面为resnet50.prototxt中数据预处理,可以看到crop_size裁剪尺寸为224,但是原始图片中有的图片尺寸小于224,所以报错了。

transform_param {
    mirror: false
    crop_size: 224
    mean_value: 104
    mean_value: 117
    mean_value: 123
  }

解决办法

因为model一旦训练好,结构是固定的,无法再调整,所以这里只能改变数据集。

# CAFFE_ROOT为caffe安装目录
$ cd $CAFFE_ROOT
$ vim examples/imagenet/create_imagenet.sh

# 修改RESIZE=false为RESIZE=true

改RESIZE_HEIGHT和RESIZE_WIDTH值的大小大于crop_size即可

RESIZE_HEIGHT=256
RESIZE_WIDTH=256

最后重新生成数据

# 备份原始数据
# 删除原始数据目录
# 生成新数据
$ examples/imagenet/create_imagenet.sh
深度学习caffe
朗读
赞(0)
赞赏
感谢您的支持,我会继续努力哒!

三合一收款

下面三种方式都支持哦

微信
QQ
支付宝
打开支付宝/微信/QQ扫一扫,即可进行扫码打赏哦
版权属于:

半醉残影

本文链接:

https://blog.dengyb.com/archives/29/(转载时请注明本文出处及文章链接)

评论 (0)