File size: 747 Bytes
fd4b932 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
#!/bin/bash
if [[ "$#" -ne 5 ]]; then
echo "Usage: $0 <model_file> <image_path> <dataset_file> <class_file> <output_path>"
exit 1
fi
MODEL_FILE=$1
IMAGE_PATH=$2
DATASET_FILE=$3
CLASS_FILE=$4
OUTPUT_PATH=$5
IMAGE_LIST=$(cat $DATASET_FILE)
IMAGE_NUM=$(cat $DATASET_FILE | wc -l)
#prepare process bar
i=0
ICON_ARRAY=("\\" "|" "/" "-")
#create output path first
mkdir -p $OUTPUT_PATH
for IMAGE_ID in $IMAGE_LIST
do
./deeplabSegment -m $MODEL_FILE -i $IMAGE_PATH"/"$IMAGE_ID".jpg" -l $CLASS_FILE -k $OUTPUT_PATH"/"$IMAGE_ID".png" -t 4 -c 1 -w 1 -p 0 2>&1 >> /dev/null
#update process bar
let index=i%4
printf "inference process: %d/%d [%c]\r" "$i" "$IMAGE_NUM" "${ICON_ARRAY[$index]}"
let i=i+1
done
printf "\nDone\n"
|