首先生成train.txt,val.txt。里面的内容就是(图片路径,标签).
首先把$CAFFE_ROOT/examples/imagenet目录下的东西全复制过来
然后,vi create_imagenet.sh,改改里面的路径。
EXAMPLE=/home/ds/PycharmProjects/mnist
DATA=/home/ds/PycharmProjects/mnist
TOOLS=/home/ds/git/caffe/build/tools
TRAIN_DATA_ROOT=/home/ds/PycharmProjects/mnist/traindata/
VAL_DATA_ROOT=/home/ds/PycharmProjects/mnist/valdata/
如果要改变图片的大小,
RESIZE=true
if $RESIZE; then
RESIZE_HEIGHT=128
RESIZE_WIDTH=128
else
RESIZE_HEIGHT=0
RESIZE_WIDTH=0
fi
然后./create_imagenet.sh就可以了。
但是如果是灰度图,就./create_imagenet.sh –gray=True.