mmocr/tests/test_models
LvTianlei 8f377f2fa2
[Feature] Add Satrn (#405)
* Add SATRN

* Create satrn_small_academic.py

* Update README.md

* change config name

* Update mmocr/models/textrecog/backbones/shallow_cnn.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update configs/textrecog/satrn/satrn_academic.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update configs/textrecog/satrn/satrn_small.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update shallow_cnn.py

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update test_ocr_encoder.py

* change keep_aspect_ratio=False

* Update transformer_layer.py

* Update configs/textrecog/satrn/satrn_small.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update configs/textrecog/satrn/satrn_academic.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update transformer_layer.py

* Apply suggestions from code review

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update transformer_layer.py

* update satrn readme

* add satrn to ocr.py

* add satrn_sm and fix configs

* add a test for config

* add copyright info

* use mmocr registry

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2021-08-19 22:02:58 +08:00
..
test_label_convertor [Enhancement] Add copyright info (#439) 2021-08-17 17:39:30 +08:00
test_detector.py [Enhancement] Add copyright info (#439) 2021-08-17 17:39:30 +08:00
test_kie_config.py [Enhancement] Add copyright info (#439) 2021-08-17 17:39:30 +08:00
test_loss.py [Enhancement] Add copyright info (#439) 2021-08-17 17:39:30 +08:00
test_modules.py [Enhancement] Add copyright info (#439) 2021-08-17 17:39:30 +08:00
test_ner_model.py [Enhancement] Add copyright info (#439) 2021-08-17 17:39:30 +08:00
test_ocr_backbone.py [Feature] Add Satrn (#405) 2021-08-19 22:02:58 +08:00
test_ocr_decoder.py [Enhancement] Add copyright info (#439) 2021-08-17 17:39:30 +08:00
test_ocr_encoder.py [Feature] Add Satrn (#405) 2021-08-19 22:02:58 +08:00
test_ocr_head.py [Enhancement] Add copyright info (#439) 2021-08-17 17:39:30 +08:00
test_ocr_layer.py [Enhancement] Add copyright info (#439) 2021-08-17 17:39:30 +08:00
test_ocr_loss.py [Enhancement] Add copyright info (#439) 2021-08-17 17:39:30 +08:00
test_ocr_neck.py [Enhancement] Add copyright info (#439) 2021-08-17 17:39:30 +08:00
test_ocr_preprocessor.py [Enhancement] Add copyright info (#439) 2021-08-17 17:39:30 +08:00
test_panhead.py [Enhancement] Add copyright info (#439) 2021-08-17 17:39:30 +08:00
test_recog_config.py [Feature] Add Satrn (#405) 2021-08-19 22:02:58 +08:00
test_recognizer.py [Enhancement] Add copyright info (#439) 2021-08-17 17:39:30 +08:00
test_targets.py [Enhancement] Add copyright info (#439) 2021-08-17 17:39:30 +08:00
test_textdet_head.py [Enhancement] Add copyright info (#439) 2021-08-17 17:39:30 +08:00
test_textdet_neck.py [Enhancement] Add copyright info (#439) 2021-08-17 17:39:30 +08:00