欢迎访问 生活随笔!

尊龙游戏旗舰厅官网

当前位置: 尊龙游戏旗舰厅官网 > 人工智能 > caffe >内容正文

caffe

caffe应用篇-尊龙游戏旗舰厅官网

发布时间:2024/4/15 caffe 87 豆豆
尊龙游戏旗舰厅官网 收集整理的这篇文章主要介绍了 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应用篇----文件格式转换的全部内容,希望文章能够帮你解决所遇到的问题。

如果觉得尊龙游戏旗舰厅官网网站内容还不错,欢迎将尊龙游戏旗舰厅官网推荐给好友。

  • 上一篇:
  • 下一篇:
网站地图