add slanet to benckmark
parent
3093f2410e
commit
d76b7f8b80
|
@ -25,7 +25,6 @@ if [ ${MODE} = "benchmark_train" ];then
|
|||
array=(${python_name_list})
|
||||
python_name=${array[0]}
|
||||
${python_name} -m pip install -r requirements.txt
|
||||
${python_name} -m pip install git+https://github.com/LDOUBLEV/AutoLog
|
||||
if [[ ${model_name} =~ "ch_ppocr_mobile_v2_0_det" || ${model_name} =~ "det_mv3_db_v2_0" ]];then
|
||||
wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/pretrained/MobileNetV3_large_x0_5_pretrained.pdparams --no-check-certificate
|
||||
rm -rf ./train_data/icdar2015
|
||||
|
@ -101,6 +100,13 @@ if [ ${MODE} = "benchmark_train" ];then
|
|||
ln -s ./pubtabnet_benckmark ./pubtabnet
|
||||
cd ../
|
||||
fi
|
||||
if [[ ${model_name} == "slanet" ]];then
|
||||
rm -rf ./train_data/pubtabnet
|
||||
wget -nc -P ./train_data/ https://paddleocr.bj.bcebos.com/dataset/pubtabnet_benckmark.tar --no-check-certificate
|
||||
cd ./train_data/ && tar xf pubtabnet_benckmark.tar
|
||||
ln -s ./pubtabnet_benckmark ./pubtabnet
|
||||
cd ../
|
||||
fi
|
||||
if [[ ${model_name} == "det_r50_dcn_fce_ctw_v2_0" ]]; then
|
||||
wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/contribution/det_r50_dcn_fce_ctw_v2.0_train.tar --no-check-certificate
|
||||
cd ./pretrain_models/ && tar xf det_r50_dcn_fce_ctw_v2.0_train.tar && cd ../
|
||||
|
|
Loading…
Reference in New Issue