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应用篇----文件格式转换的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。