#!/usr/bin/env sh

# Create the imagenet lmdb inputs

# N.B. set the path to the imagenet train + val data dirs

EXAMPLE=examples/imagenet  #可以改成自己文件夹名

DATA=data/ilsvrc12  #txt文件存储的位置

TOOLS=build/tools

TRAIN_DATA_ROOT=/path/to/imagenet/train/  #要改成自己文件的路径

VAL_DATA_ROOT=/path/to/imagenet/val/

# Set RESIZE=true to resize the images to 256x256. Leave as false if images have

# already been resized using another tool.

RESIZE=false

if $RESIZE; then

  RESIZE_HEIGHT=

  RESIZE_WIDTH=

else

  RESIZE_HEIGHT=

  RESIZE_WIDTH=

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

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

fi

echo "Creating train lmdb..."

GLOG_logtostderr= $TOOLS/convert_imageset \

    --resize_height=$RESIZE_HEIGHT \

    --resize_width=$RESIZE_WIDTH \

    --shuffle \

    $TRAIN_DATA_ROOT \

    $DATA/train.txt \   # #改成自己文件的训练集txt名

    $EXAMPLE/ilsvrc12_train_lmdb  #改成自己文件夹的名称，不改的话则自动生成

echo "Creating val lmdb..."

GLOG_logtostderr= $TOOLS/convert_imageset \

    --resize_height=$RESIZE_HEIGHT \

    --resize_width=$RESIZE_WIDTH \

    --shuffle \

    $VAL_DATA_ROOT \

    $DATA/val.txt \   #改成自己文件的测试集txt名

    $EXAMPLE/ilsvrc12_val_lmdb  #改成自己文件夹的名称，不改的话则自动生成

echo "Done."