caffe生成lmdb数据集的脚本

最新推荐文章于 2021-09-28 10:39:22 发布

半城烟雨半城花

最新推荐文章于 2021-09-28 10:39:22 发布

阅读量487

点赞数

CC 4.0 BY-SA版权

分类专栏：深度学习 caffe 制作数据集文章标签：制作数据集 caffe 深度学习

本文链接：https://blog.youkuaiyun.com/weixin_43439673/article/details/83108521

caffe 同时被 3 个专栏收录

11 篇文章

订阅专栏

深度学习

9 篇文章

订阅专栏

制作数据集

1 篇文章

订阅专栏

本文详细介绍如何使用Caffe框架的convert_imageset工具，将ImageNet数据集转换为LMDB格式，适用于训练和验证阶段。文章包括设置环境变量、检查数据路径及执行转换命令等关键步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

caffe生成lmdb数据集的脚本

#!/usr/bin/env sh
set -e

EXAMPLE=/home/hadoop/桌面/caffe_study/more_eyes
DATA=/home/hadoop/桌面/caffe_study/more_eyes
TOOLS=/home/hadoop/caffe/build/tools

TRAIN_DATA_ROOT=/home/hadoop/桌面/caffe_study/more_eyes/train/
VAL_DATA_ROOT=/home/hadoop/桌面/caffe_study/more_eyes/test/


RESIZE=true
if $RESIZE; then
  RESIZE_HEIGHT=32
  RESIZE_WIDTH=32
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 lmdb..."

GLOG_logtostderr=1 $TOOLS/convert_imageset \
    --resize_height=$RESIZE_HEIGHT \
    --resize_width=$RESIZE_WIDTH \
    --shuffle \
    $TRAIN_DATA_ROOT \
    $DATA/train.txt \
    $EXAMPLE/train_lmdb

echo "Creating val lmdb..."

GLOG_logtostderr=1 $TOOLS/convert_imageset \
    --resize_height=$RESIZE_HEIGHT \
    --resize_width=$RESIZE_WIDTH \
    --shuffle \
    $VAL_DATA_ROOT \
    $DATA/test.txt \
    $EXAMPLE/test_lmdb

echo "Done."

有问题欢迎评论,我们共同解决

喜欢请关注,转载请注明出处,谢谢配合