更换文档检测模型
This commit is contained in:
23
paddle_detection/deploy/benchmark/benchmark_quant.sh
Normal file
23
paddle_detection/deploy/benchmark/benchmark_quant.sh
Normal file
@@ -0,0 +1,23 @@
|
||||
# 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
|
||||
|
||||
Reference in New Issue
Block a user