caffe应用篇-ag凯发k8国际

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

总结

以上是ag凯发k8国际为你收集整理的caffe应用篇----文件格式转换的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得ag凯发k8国际网站内容还不错，欢迎将ag凯发k8国际推荐给好友。