https://github.com/mountchicken/ctpn_crnn_chineseocr_pyqt5

CTPN and CRNN based Chinese OCR, developed with PyQt5
https://github.com/mountchicken/ctpn_crnn_chineseocr_pyqt5

chinese crnn ctpn gui ocr pyqt5 pytorch

Last synced: 5 days ago
JSON representation

CTPN and CRNN based Chinese OCR, developed with PyQt5

• Host: GitHub
• URL: https://github.com/mountchicken/ctpn_crnn_chineseocr_pyqt5
• Owner: Mountchicken
• Created: 2021-04-18T03:30:36.000Z (over 3 years ago)
• Default Branch: main
• Last Pushed: 2021-09-18T08:35:01.000Z (about 3 years ago)
• Last Synced: 2024-11-02T08:51:29.131Z (12 days ago)
• Topics: chinese, crnn, ctpn, gui, ocr, pyqt5, pytorch
• Language: Python
• Homepage:
• Size: 29.9 MB
• Stars: 22
• Watchers: 1
• Forks: 3
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# CTPN_CRNN_ChineseOCR_PyQt5

CTPN and CRNN based Chinese OCR, developed with PyQt5

# Examples

- Hello guys, hope you are doing fine these days !😄

- In this repositories, i created a PyQt5 Application to do some Chinese OCR job which is based on CTPN and CRNN

- Here is the result, hope you enjoy it

- However, the CRNN model doesn't work well(Terrible at some situation😥). You will find it out when you try 

- ![test_example](https://github.com/Mountchicken/CTPN_CRNN_ChineseOCR_PyQt5/blob/main/github/detectd.JPG)

# Requirements

- PyQt5

- torch

- torchvision

# How to use it

- I'm sorry guys, i haven't find a way to deploy it, and you have to run it in your compiler 🙇‍♂️(VScode, pycharm or...)

## Build Environment

```python

cd CTPN_lib/bbox

python setup.py build

```

- a file named build will be generated, put `bbox.cp37-win_amd64.pyd` and `nms.cp37-win_amd64.pyd` CTPN_lib/bbox

## download pretrained weights

- The weights are larger than the uploading limit(25M below😅). Download them using BaiduYun

- Put them in CTPN_weights [CTPN weights(提取码:r88t)](https://pan.baidu.com/s/1RGzKOF3efErMTtpzry-hYQ)

- Put them in CRNN_weights [CRNN weights(提取码:tnts)](https://pan.baidu.com/s/1kLUhx-zI9BeF4rTNC9u2Bg)

## Choose which model to use

- In `inferrence.py`, from line 27 to line 32

- `argument: crnn_weights`: the file location of crnn weigth downloaded in the previous step

- `argument: ctpn_basemodel`:choose a ctpb backbone: vgg16, resnet50, shufflenet_v2_x1_0, mobilenet_v3_large, mobilenet_v3_small

- `argument: ctpn_weights`:corresponding ctpn weights with ctpn_base model downloaded in the previous step

## Run main_gui.py

- if you run the .py file succesfully, it should look like this

- ![test_example](https://github.com/Mountchicken/CTPN_CRNN_ChineseOCR_PyQt5/blob/main/github/menu.JPG)

- Then, you need to push the initialize button to load the model, after that, just wait the `Finished` sign appers in the right.

- ![test_example](https://github.com/Mountchicken/CTPN_CRNN_ChineseOCR_PyQt5/blob/main/github/Initialized.JPG)

- Finally, load the image with `Load Image` button and press `Detect`

- ![test_example](https://github.com/Mountchicken/CTPN_CRNN_ChineseOCR_PyQt5/blob/main/github/detectd.JPG)

# For more issue, contact me

- `Email Address` [email protected]