caffe应用篇-尊龙游戏旗舰厅官网
1 训练前准备
caffe常使用的数据是db格式(leveldb/lmdb),我们手中有的图片数据,jpg、bmp格式等,必须转换;此操作caffe已经提供工具。它的路径是/home/your/caffe-master/build/tools/convert_imageset.bin;不过,使用它之前,需要先生成一个依赖文本文件。
convert_imageset.bin程序的使用格式:
convert_imageset [flags] rootfolder/ listfile db_name需要带四个参数:
flags: 图片参数组,后面详细介绍
rootfolder/: 图片存放的绝对路径,从linux系统根目录开始
listfile: 图片文件列表清单,一般为一个txt文件,一行一张图片
db_name: 最终生成的db文件存放目录
2 训练文本listfile生成
用gedit编辑如下shell代码,并存为mktraintxt.sh# /usr/bin/env sh data=examples/images echo "create train.txt..." rm -rf $data/train.txt find $data -name *cat.jpg | cut -d '/' -f3 | sed "s/$/ 1/">>$data/train.txt find $data -name *bike.jpg | cut -d '/' -f3 | sed "s/$/ 2/">>$data/tmp.txt cat $data/tmp.txt>>$data/train.txt rm -rf $data/tmp.txt echo "done.."
3 图像转换代码 sudo vim data/mydata/create_lmdb.sh(我的数据放在data文件下的mydata目录下)。
#!/usr/bin/env sh
# this script converts the mnist data into lmdb/leveldb format,
# depending on the value assigned to $backend.
set -e
example=examples/mnist
data=data/mnist
build=build/examples/mnist
backend="lmdb"
train_data_root=/home/yan/image-face/train/
val_data_root=/home/yan/image-face/val/
resize=true
if $resize; then
resize_height=227
resize_width=227
else
resize_height=0
resize_width=0
fi
if [ ! -d "$train_data_root" ]; then
echo "error: train_data_root is not a path to a directory: $train_data_root"
echo "set the train_data_root variable in create_imagenet.sh to the path" \
"where the imagenet training data is stored."
exit 1
fi
if [ ! -d "$val_data_root" ]; then
echo "error: val_data_root is not a path to a directory: $val_data_root"
echo "set the val_data_root variable in create_imagenet.sh to the path" \
"where the imagenet validation data is stored."
exit 1
fi
echo "creating train ${backend}..."
glog_logtostderr=1 $tools/convert_imageset \
--resize_height=$resize_height \
--resize_width=$resize_width \
--shuffle \
$train_data_root \
$data/train_all.txt \
$example/face_train_lmdb
echo "creating val ${backend}..."
glog_logtostderr=1 $tools/convert_imageset \
--resize_height=$resize_height \
--resize_width=$resize_width \
--shuffle \
$val_data_root \
$data/val.txt \
$example/face_val_lmdb
echo "done."
转载于:https://www.cnblogs.com/gongdiwudu/p/6240190.html
总结
以上是尊龙游戏旗舰厅官网为你收集整理的caffe应用篇----文件格式转换的全部内容,希望文章能够帮你解决所遇到的问题。
- 上一篇:
- 下一篇: