forked from PulseFocusPlatform/PulseFocusPlatform
24 lines
558 B
Bash
24 lines
558 B
Bash
|
# 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 \
|
||
|
|
--use_gpu=True \
|
||
|
|
--run_mode=${run_mode} \
|
||
|
|
--image_dir=${img_dir} 2>&1 | tee ${log_path}/${model_name}_gpu_runmode_${run_mode}_bs1_infer.log
|
||
|
|
done
|
||
|
|
|