【文件属性】:
文件名称:DBnet-lite.pytorch:具有差分二值化的实时场景文本检测的pytorch重新实现
文件大小:601KB
文件格式:ZIP
更新时间:2021-04-30 15:21:05
lightweight ocr python3 pytorch dbnet
| 简体中文
DBNet-lite-pytorch
这个项目之后会在这里更新,我把之前的项目都做了下整合
环境配置
pip install -r requirement.txt
cd models/dcn/
sh make.sh
水平或倾斜文本格式
照着icdar2015的格式, x1,y1,x2,y2,x3,y3,x4,y4,label,
image
│ 1.jpg
│ 2.jpg
│ ...
label
│ gt_1.txt
│ gt_2.txt
| ...
弧形文本的格式
数据格式, x1,y1,x2,y2,x3,y3,x4,y4 ...xn,yn,label
n个点组成,n的个数可以不定
image
│ 1.jpg
│ 2.jpg
│ ...
label
│ gt_1.txt
│ gt_2.txt
| ...
训练部分
在根目录的
【文件预览】:
DBnet-lite.pytorch-master
----models()
--------head()
--------__pycache__()
--------DBNet.py(1KB)
--------__init__.py(162B)
--------backbone()
--------dcn()
----pruned()
--------prune.py(9KB)
--------prune_inference.py(4KB)
--------__init__.py(162B)
--------get_pruned_model.py(6KB)
--------train_fintune.py(9KB)
----cal_rescall()
--------rrc_evaluation_funcs.py(16KB)
--------__pycache__()
--------__init__.py(0B)
--------script.py(14KB)
----show()
--------2.jpg(156KB)
--------1.jpg(329KB)
----config.yaml(2KB)
----inference.py(4KB)
----loss()
--------loss.py(1KB)
--------balance_cross_entropy_loss.py(2KB)
--------__pycache__()
--------__init__.py(162B)
--------dice_loss.py(7KB)
--------l1_loss.py(1KB)
----train.py(8KB)
----dataloader()
--------random_thansform.py(9KB)
--------__pycache__()
--------MakeBorderMap.py(5KB)
--------MakeSegMap.py(3KB)
--------__init__.py(162B)
--------dataload.py(4KB)
----README_en.md(2KB)
----requirement.txt(191B)
----README.md(3KB)
----utils()
--------Logger.py(3KB)
--------tools.py(3KB)
--------metrics.py(2KB)
--------__pycache__()
--------fuse_model.py(2KB)
--------set_optimizer.py(753B)
--------__init__.py(162B)
--------model_eval.py(4KB)
--------DB_postprocesss.py(8KB)