add convert_imageset option to resize images; use in

convert_imageset.cpp and document

1 files changed, 37 insertions, 5 deletions

diff --git a/examples/imagenet/create_imagenet.sh b/examples/imagenet/create_imagenet.sh
index a1bcb7bf..8767f120 100755
--- a/examples/imagenet/create_imagenet.sh
+++ b/examples/imagenet/create_imagenet.sh
@@ -5,16 +5,48 @@
 TOOLS=../../build/tools
 DATA=../../data/ilsvrc12
 
-echo "Creating leveldb..."
+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=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 leveldb..."
 
 GLOG_logtostderr=1 $TOOLS/convert_imageset.bin \
-    /path/to/imagenet/train/ \
+    $TRAIN_DATA_ROOT \
     $DATA/train.txt \
-    imagenet_train_leveldb 1
+    imagenet_train_leveldb 1 \
+    $RESIZE_HEIGHT $RESIZE_WIDTH
+
+echo "Creating val leveldb..."
 
 GLOG_logtostderr=1 $TOOLS/convert_imageset.bin \
-    /path/to/imagenet/val/ \
+    $VAL_DATA_ROOT \
     $DATA/val.txt \
-    imagenet_val_leveldb 1
+    imagenet_val_leveldb 1 \
+    $RESIZE_HEIGHT $RESIZE_WIDTH
 
 echo "Done."