文件名称:DBnet-lite.pytorch:具有差分二值化的实时场景文本检测的pytorch重新实现
文件大小:601KB
文件格式:ZIP
更新时间:2024-05-20 09:07:45
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)