| 1234567891011121314151617181920212223 |
- # All rights `PaddleDetection` reserved
- #!/bin/bash
- model_dir=$1
- model_name=$2
- export img_dir="demo"
- export log_path="output_pipeline"
- echo "model_dir : ${model_dir}"
- echo "img_dir: ${img_dir}"
- # TODO: support batch size>1
- for run_mode in "trt_int8"; do
- echo "${model_name} ${model_dir}, run_mode: ${run_mode}"
- python deploy/python/infer.py \
- --model_dir=${model_dir} \
- --run_benchmark=True \
- --device=GPU \
- --run_mode=${run_mode} \
- --image_dir=${img_dir} 2>&1 | tee ${log_path}/${model_name}_gpu_runmode_${run_mode}_bs1_infer.log
- done
|
