diff --git a/benchmark/run_det.sh b/benchmark/run_det.sh index 68109b3ab2c3b8b61a0c90b4b31fd855c1ba2d46..1bbe7f5bcb339308ead6ca75d83561eb9c69fdba 100644 --- a/benchmark/run_det.sh +++ b/benchmark/run_det.sh @@ -5,6 +5,7 @@ python3.7 -m pip install -r requirements.txt # 2 拷贝该模型需要数据、预训练模型 wget -P ./train_data/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/test/icdar2015.tar && cd train_data && tar xf icdar2015.tar && cd ../ wget -P ./pretrain_models/ https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/ResNet50_vd_pretrained.pdparams +wget -P ./pretrain_models/ https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/ResNet18_vd_pretrained.pdparams # 3 批量运行(如不方便批量,1,2需放到单个模型中) model_mode_list=(det_res18_db_v2.0 det_r50_vd_east det_r50_vd_pse)