Commit 4da1c86f authored by Mahmoud Lababidi's avatar Mahmoud Lababidi Committed by Neal Wu
Browse files

fix missing build images script

parent dd983ea6
...@@ -44,7 +44,8 @@ DATA_DIR="${1%/}" ...@@ -44,7 +44,8 @@ DATA_DIR="${1%/}"
SCRATCH_DIR="${DATA_DIR}/raw-data/" SCRATCH_DIR="${DATA_DIR}/raw-data/"
mkdir -p "${DATA_DIR}" mkdir -p "${DATA_DIR}"
mkdir -p "${SCRATCH_DIR}" mkdir -p "${SCRATCH_DIR}"
WORK_DIR="$0.runfiles/inception/inception" # http://stackoverflow.com/questions/59895/getting-the-source-directory-of-a-bash-script-from-within
WORK_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
# Download the flowers data. # Download the flowers data.
DATA_URL="http://download.tensorflow.org/example_images/flower_photos.tgz" DATA_URL="http://download.tensorflow.org/example_images/flower_photos.tgz"
...@@ -87,9 +88,9 @@ done < "${LABELS_FILE}" ...@@ -87,9 +88,9 @@ done < "${LABELS_FILE}"
# Build the TFRecords version of the image data. # Build the TFRecords version of the image data.
cd "${CURRENT_DIR}" cd "${CURRENT_DIR}"
BUILD_SCRIPT="${WORK_DIR}/build_image_data" BUILD_SCRIPT="${WORK_DIR}/build_image_data.py"
OUTPUT_DIRECTORY="${DATA_DIR}" OUTPUT_DIRECTORY="${DATA_DIR}"
"${BUILD_SCRIPT}" \ python "${BUILD_SCRIPT}" \
--train_directory="${TRAIN_DIRECTORY}" \ --train_directory="${TRAIN_DIRECTORY}" \
--validation_directory="${VALIDATION_DIRECTORY}" \ --validation_directory="${VALIDATION_DIRECTORY}" \
--output_directory="${OUTPUT_DIRECTORY}" \ --output_directory="${OUTPUT_DIRECTORY}" \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment