p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

squeezenet amend 310ifer shell script

This commit is contained in:
chenweitao_295 2021-05-31 19:03:42 +08:00
parent 20b1e0291c
commit 59631fca6e
2 changed files with 29 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/cv/squeezenet/README.md
View File

@ -362,7 +362,7 @@ Before performing inference, the mindir file must be exported by `export.py` scr
bash run_infer_310.sh [MINDIR_PATH] [DATASET] [DATA_PATH] [LABEL_PATH] [DEVICE_ID]
```
- `DATASET` should be in ["imagenet", "cifar10"].
- `DATASET` should be in ["imagenet", "cifar10"]. If the DATASET is cifar10, you don't need to set LABEL_FILE.
- `LABEL_PATH` label.txt path, LABEL_FILE is only useful for imagenet. Write a py script to sort the category under the dataset, map the file names under the categories and category sort values,Such as[file name : sort value], and write the mapping results to the labe.txt file.
- `DEVICE_ID` is optional, default value is 0.

35
model_zoo/official/cv/squeezenet/scripts/run_infer_310.sh
View File

@ -14,9 +14,10 @@
# limitations under the License.
# ============================================================================
if [[ $# -lt 4 || $# -gt 5 ]]; then
if [[ $# -lt 3 || $# -gt 5 ]]; then
echo "Usage: bash run_infer_310.sh [MINDIR_PATH] [DATASET] [DATA_PATH] [LABEL_FILE] [DEVICE_ID]
DEVICE_ID is optional, default value is zero, LABEL_FILE is only useful for imagenet "
LABEL_FILE is only useful for imagenet. If the DATASET is cifar10, you don't need to set LABEL_FILE.
DEVICE_ID is optional, default value is zero. "
exit 1
fi
@ -27,14 +28,34 @@ get_real_path(){
echo "$(realpath -m $PWD/$1)"
fi
}
model=$(get_real_path $1)
dataset=$2
data_path=$(get_real_path $3)
label_file=$(get_real_path $4)
device_id=0
if [ $# == 5 ]; then
device_id=$5
if [ "x${dataset}" == "xcifar10" ] || [ "x${dataset}" == "xCifar10" ]; then
if [[ $# -gt 4 ]]; then
echo "Usage: bash run_infer_310.sh [MINDIR_PATH] [DATASET] [DATA_PATH] [LABEL_FILE] [DEVICE_ID]
LABEL_FILE is only useful for imagenet. If the DATASET is cifar10, you don't need to set LABEL_FILE.
DEVICE_ID is optional, default value is zero. "
exit 1
fi
label_file=./preprocess_Result/label
device_id=0
if [ $# == 4 ]; then
device_id=$4
fi
else
if [[ $# -lt 4 ]]; then
echo "Usage: bash run_infer_310.sh [MINDIR_PATH] [DATASET] [DATA_PATH] [LABEL_FILE] [DEVICE_ID]
LABEL_FILE is only useful for imagenet. If the DATASET is cifar10, you don't need to set LABEL_FILE.
DEVICE_ID is optional, default value is zero. "
exit 1
fi
label_file=$(get_real_path $4)
device_id=0
if [ $# == 5 ]; then
device_id=$5
fi
fi
echo "mindir name: "$model