|
#!/usr/bin/env sh |
|
|
|
|
|
set -e |
|
|
|
EXAMPLE=examples/imagenet |
|
DATA=data/ilsvrc12 |
|
TOOLS=build/tools |
|
|
|
TRAIN_DATA_ROOT=/path/to/imagenet/train/ |
|
VAL_DATA_ROOT=/path/to/imagenet/val/ |
|
|
|
|
|
|
|
RESIZE=false |
|
if $RESIZE; then |
|
RESIZE_HEIGHT=256 |
|
RESIZE_WIDTH=256 |
|
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/ilsvrc12_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/val.txt \ |
|
$EXAMPLE/ilsvrc12_val_lmdb |
|
|
|
echo "Done." |
|
|