sayybro / AGC2022_Task3_OCR

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Drone_Task3_OCR

  1. Download Pretrained model from CRAFT and WIW github.
  2. python task3.py --craft_weight ./craft_mlt_25k.pth --wiw_weight ./recognition_model.pth

We use the pretrained model for Detection and finetuned model for Recognition.

[STAGE 1] Detection: CRAFT

Baek, Youngmin, et al. "Character region awareness for text detection." Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. 2019.

[STAGE 2] Recognition: WIW

Baek, Jeonghun, et al. "What is wrong with scene text recognition model comparisons? dataset and model analysis." Proceedings of the IEEE/CVF international conference on computer vision. 2019.

About


Languages

Language:Python 100.0%