remove stn in svtr

eshoyuan · Sep 8, 2022 · 8a75ed5 · 8a75ed5
1 parent 1005aaa
commit 8a75ed5
Show file tree

Hide file tree

Showing 6 changed files with 163 additions and 369 deletions.
diff --git a/preprocess.sh b/preprocess.sh
@@ -2,4 +2,6 @@ python3 crop.py
 python3 create_classification_dataset.py
 python3 classification_vgg.py
 python3 create_PSOL_dataset.py
-python3 PSOL/generate_box_imagenet_crop.py
+python3 PSOL/generate_box_imagenet_crop.py
+python3 tools/train.py -c svtr_large_train_stn.yml # Distributed Training: python -m paddle.distributed.launch --gpus '0,1,2,3' -c svtr_large_train_stn.yml
+python3 tools/train.py -c svtr_large_train_stn.yml -o Global.pretrained_model=svtr_large_stn/best_accuracy
diff --git a/rec_PP-OCRv3.yml b/rec_PP-OCRv3.yml
diff --git a/rec_svtrnet.yml b/rec_svtrnet.yml
diff --git a/requirements.txt b/requirements.txt
@@ -15,4 +15,5 @@ cython
 lxml
 premailer
 openpyxl
-attrdict
+attrdict
+pyyaml
diff --git a/svtr_large.yml b/svtr_large.yml