diff --git a/models/cv/classification/densenet161/ixrt/README.md b/models/cv/classification/densenet161/ixrt/README.md index 08e7bd16deccf92eb7868713e7dd08768307e9d5..cb65f64a23538a7d67ac3ee4fbb171cd89c427c6 100644 --- a/models/cv/classification/densenet161/ixrt/README.md +++ b/models/cv/classification/densenet161/ixrt/README.md @@ -31,21 +31,21 @@ Dataset: to download the validation dat ```bash python3 export.py --weight densenet161-8d451a50.pth --output densenet161.onnx -cd data && mkdir checkpoints && cd checkpoints && mkdir densenet161 -mv densenet161.onnx densenet161 ``` ## Inference +```bash +export DATASETS_DIR=/Path/to/imagenet_val/ +``` ### FP16 ```bash -cd deepsparkinference # Accuracy -bash models/cv/classification/densenet161/ixrt/scripts/infer_densenet_fp16_accuracy.sh +bash scripts/infer_densenet_fp16_accuracy.sh # Performance -bash models/cv/classification/densenet161/ixrt/scripts/infer_densenet_fp16_performance.sh +bash scripts/infer_densenet_fp16_performance.sh ``` ## Results diff --git a/models/cv/classification/densenet161/ixrt/scripts/infer_densenet_fp16_accuracy.sh b/models/cv/classification/densenet161/ixrt/scripts/infer_densenet_fp16_accuracy.sh index e266c65a5c4981b1883765846d8b050d38d96050..dfaf40d720078b6af9b6d98410118bfbda894b09 100644 --- a/models/cv/classification/densenet161/ixrt/scripts/infer_densenet_fp16_accuracy.sh +++ b/models/cv/classification/densenet161/ixrt/scripts/infer_densenet_fp16_accuracy.sh @@ -42,11 +42,11 @@ do esac done -PROJ_DIR=$(cd $(dirname $0);cd ../../../../../..; pwd) -DATASETS_DIR="${PROJ_DIR}/data/datasets/imagenet_val/" -CHECKPOINTS_DIR="${PROJ_DIR}/data/checkpoints/densenet161/" -RUN_DIR="${PROJ_DIR}/models/cv/classification/densenet161/ixrt/" -CONFIG_DIR="${PROJ_DIR}/models/cv/classification/densenet161/ixrt/config/DENSENET_CONFIG" +PROJ_DIR=${PROJ_DIR:-"."} +DATASETS_DIR="${DATASETS_DIR}" +CHECKPOINTS_DIR="${PROJ_DIR}" +RUN_DIR="${PROJ_DIR}" +CONFIG_DIR="${PROJ_DIR}/config/DENSENET_CONFIG" source ${CONFIG_DIR} ORIGINE_MODEL=${CHECKPOINTS_DIR}/${ORIGINE_MODEL} diff --git a/models/cv/classification/densenet161/ixrt/scripts/infer_densenet_fp16_performance.sh b/models/cv/classification/densenet161/ixrt/scripts/infer_densenet_fp16_performance.sh index cc0816efbacd92e2fb81945dc8e01f39fbdaee83..f09ccfcc070c0a382212671650184efa4432da28 100644 --- a/models/cv/classification/densenet161/ixrt/scripts/infer_densenet_fp16_performance.sh +++ b/models/cv/classification/densenet161/ixrt/scripts/infer_densenet_fp16_performance.sh @@ -42,11 +42,11 @@ do esac done -PROJ_DIR=$(cd $(dirname $0);cd ../../../../../..; pwd) -DATASETS_DIR="${PROJ_DIR}/data/datasets/imagenet_val/" -CHECKPOINTS_DIR="${PROJ_DIR}/data/checkpoints/densenet161/" -RUN_DIR="${PROJ_DIR}/models/cv/classification/densenet161/ixrt/" -CONFIG_DIR="${PROJ_DIR}/models/cv/classification/densenet161/ixrt/config/DENSENET_CONFIG" +PROJ_DIR=${PROJ_DIR:-"."} +DATASETS_DIR="${DATASETS_DIR}" +CHECKPOINTS_DIR="${PROJ_DIR}" +RUN_DIR="${PROJ_DIR}" +CONFIG_DIR="${PROJ_DIR}/config/DENSENET_CONFIG" source ${CONFIG_DIR} ORIGINE_MODEL=${CHECKPOINTS_DIR}/${ORIGINE_MODEL}