https://github.com/jahongir7174/dbnet
Real-time Scene Text Detection with Differentiable Binarization
https://github.com/jahongir7174/dbnet
python pytorch testing text-detection training
Last synced: 7 months ago
JSON representation
Real-time Scene Text Detection with Differentiable Binarization
- Host: GitHub
- URL: https://github.com/jahongir7174/dbnet
- Owner: jahongir7174
- Created: 2024-03-21T06:31:54.000Z (over 1 year ago)
- Default Branch: master
- Last Pushed: 2024-04-26T00:24:28.000Z (over 1 year ago)
- Last Synced: 2025-01-19T00:50:11.367Z (9 months ago)
- Topics: python, pytorch, testing, text-detection, training
- Language: Python
- Homepage: https://arxiv.org/abs/1911.08947
- Size: 85.4 MB
- Stars: 1
- Watchers: 0
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
[Real-time Scene Text Detection with Differentiable Binarization](https://arxiv.org/abs/1911.08947)
### Installation
```
conda create -n PyTorch python=3.8
conda activate PyTorch
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch-lts
pip install opencv-python==4.5.5.64
pip install pyyaml
pip install timm
pip install tqdm
pip install shapely
pip install pyclipper
```
### Train
* Configure your dataset path in `main.py` for training
* Run `bash main.sh $ --train` for training, `$` is number of GPUs
### Test
* Configure your path in `main.py` for testing
* Run `python main.py --test` for testing
### Demo
* Configure your path in `main.py` for demo
* Run `python main.py --demo` for demo
### Results
| Version | Epochs | Precision | Recall | F1 | Backbone |
|:-------:|:------:|----------:|-------:|-----:|----------:|
| DB | 1200 | 83.1 | 79.3 | 81.2 | ResNet-18 |
| DB* | 1200 | 82.9 | 75.9 | 79.2 | ResNet-18 |
* `*` means that the model is trained from original repo, see reference
#### Reference
* https://github.com/MhLiao/DB