Unverified Commit 31b0e518 authored by Yukun Zhu's avatar Yukun Zhu Committed by GitHub
Browse files

Allow deeplab dataset script to be run from any location (#8466)

Thanks amangu@ for fixing this script.
parent d93a1206
...@@ -37,6 +37,7 @@ set -e ...@@ -37,6 +37,7 @@ set -e
CURRENT_DIR=$(pwd) CURRENT_DIR=$(pwd)
WORK_DIR="./pascal_voc_seg" WORK_DIR="./pascal_voc_seg"
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
mkdir -p "${WORK_DIR}" mkdir -p "${WORK_DIR}"
cd "${WORK_DIR}" cd "${WORK_DIR}"
...@@ -69,7 +70,7 @@ SEG_FOLDER="${PASCAL_ROOT}/SegmentationClass" ...@@ -69,7 +70,7 @@ SEG_FOLDER="${PASCAL_ROOT}/SegmentationClass"
SEMANTIC_SEG_FOLDER="${PASCAL_ROOT}/SegmentationClassRaw" SEMANTIC_SEG_FOLDER="${PASCAL_ROOT}/SegmentationClassRaw"
echo "Removing the color map in ground truth annotations..." echo "Removing the color map in ground truth annotations..."
python ./remove_gt_colormap.py \ python3 "${SCRIPT_DIR}/remove_gt_colormap.py" \
--original_gt_folder="${SEG_FOLDER}" \ --original_gt_folder="${SEG_FOLDER}" \
--output_dir="${SEMANTIC_SEG_FOLDER}" --output_dir="${SEMANTIC_SEG_FOLDER}"
...@@ -82,7 +83,7 @@ IMAGE_FOLDER="${PASCAL_ROOT}/JPEGImages" ...@@ -82,7 +83,7 @@ IMAGE_FOLDER="${PASCAL_ROOT}/JPEGImages"
LIST_FOLDER="${PASCAL_ROOT}/ImageSets/Segmentation" LIST_FOLDER="${PASCAL_ROOT}/ImageSets/Segmentation"
echo "Converting PASCAL VOC 2012 dataset..." echo "Converting PASCAL VOC 2012 dataset..."
python ./build_voc2012_data.py \ python3 "${SCRIPT_DIR}/build_voc2012_data.py" \
--image_folder="${IMAGE_FOLDER}" \ --image_folder="${IMAGE_FOLDER}" \
--semantic_segmentation_folder="${SEMANTIC_SEG_FOLDER}" \ --semantic_segmentation_folder="${SEMANTIC_SEG_FOLDER}" \
--list_folder="${LIST_FOLDER}" \ --list_folder="${LIST_FOLDER}" \
......
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