Переглянути джерело

fix missing build images script

Mahmoud Lababidi 8 роки тому
батько
коміт
4da1c86f94

+ 4 - 3
inception/inception/data/download_and_preprocess_flowers.sh

@@ -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}"
 
 
 # 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}" \