caffe利用shell创建train.txt和val.txt做数据输入
来源:互联网 发布:俄罗斯域名注册商 编辑:程序博客网 时间:2024/05/21 11:26
原图像数据:
首先必须熟悉一下,find,cut,sed 命令的使用。
# /usr/bin/env shDATA=examples/images #数据目录echo "Create train.txt..."rm -rf $DATA/train.txtfind $DATA -name *cat.jpg | cut -d '/' -f3 | sed "s/$/ 1/">>$DATA/train.txtfind $DATA -name *bike.jpg | cut -d '/' -f3 | sed "s/$/ 2/">>$DATA/tmp.txtcat $DATA/tmp.txt>>$DATA/train.txtrm -rf $DATA/tmp.txtecho "Done.."
在caffe根目录下运行:
$ ./examples/images/creat_picturelist.sh
运行结果:
此时,配置文件的数据输入层改:
layer { name: "Data" type: "ImageData" top: "data" top: "label" include { phase: TRAIN } transform_param { scale: 0.00390625 } image_data_param{ source:"/home/xxx/caffe/examples/train.txt" shuffle : true batch_size: 20 new_height: 141 new_width: 141 }}
打完,手工!
阅读全文