那棵树看起来生气了
[常见错误] caffe跑resnet50 imagenet数据集错误height <= datum_height
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
三合一收款
下面三种方式都支持哦