pytorch crnn训练

2024-04-17 07:19:36

一.代码下载

代码
这个代码主要是针对中文的检测。

二.环境配置

WIN 10 or Ubuntu 16.04
PyTorch 1.2.0 (may fix ctc loss) with cuda 10.0 fire
yaml
easydict
tensorboardX
解释：pytorch版本不需要严格对应，但是建议安装新版本；pytorch版本>=1.1.0，其自带ctcloss,否则需要自己安装Warp-ctc。

三.测试demo

在根目录下运行demo.py即可，会输出相应文字，测试图片在images文件夹下的test.png图像。

四.训练

1.下载数据集数据
1）对数据进行解压，.rar文件在ubuntu中解压需要安装插件。

#压缩文件为rar的安装包
sudo apt-get install rar
#解压rar文件的安装包
sudo apt-get install unrar
#解压文件
sudo unrar x name.rar

备注：该文件名字Synthetic Chinese String Dataset .rar中包含空格，解压时建议自己修改以下名字。
解压后会生成一个images的文件夹，建议不要随便点进去，360万张图像容易卡电脑。
2）打开lib/config/360CC_config.yaml文件，将第15行，ROOT这个路径修改为自己解压生成的images文件路径

2.下载标签
1）label提取码：eaqb
2）将char_std_5990.txt，train.txt 和 test.txt文件放入lib/dataset/txt/中，txt文件夹下本身有这三个文件，删除原始的。
3.训练

python train.py --cfg lib/config/360CC_config.yaml

pytorch crnn训练相关推荐

手把手教你洞悉 PyTorch 模型训练过程，彻底掌握 PyTorch 项目实战！（文末重金招聘导师）...
(文末重金招募导师) 在CVPR 2020会议接收中,PyTorch 使用了405次,TensorFlow 使用了102次,PyTorch使用数是TensorFlow的近4倍. 自2019年开始,越来 ...
python建立文件数据库_Python创建CRNN训练用的LMDB数据库文件
CRNN简介 CRNN由 Baoguang Shi, Xiang Bai, Cong Yao提出,2015年7月发表论文:"An End-to-End Trainable Neural Ne ...
PyTorch分布式训练
PyTorch分布式训练 PyTorch 是一个 Python 优先的深度学习框架,能够在强大的 GPU 加速基础上实现张量和动态神经网络.PyTorch的一大优势就是它的动态图计算特性. Licen ...
快速上手笔记，PyTorch模型训练实用教程（附代码）
机器之心发布作者:余霆嵩前言自 2017 年 1 月 PyTorch 推出以来,其热度持续上升,一度有赶超 TensorFlow 的趋势.PyTorch 能在短时间内被众多研究人员和工程师接受并 ...
新手手册：Pytorch分布式训练
文 | 花花@机器学习算法与自然语言处理单位 | SenseTime 算法研究员目录 0X01 分布式并行训练概述 0X02 Pytorch分布式数据并行 0X03 手把手渐进式实战 A. 单机单 ...
PyTorch 分布式训练DDP 单机多卡快速上手
PyTorch 分布式训练DDP 单机多卡快速上手本文旨在帮助新人快速上手最有效的 PyTorch 单机多卡训练,对于 PyTorch 分布式训练的理论介绍.多方案对比,本文不做详细介绍,有兴趣的读 ...
【资源下载】《Pytorch模型训练实现教程》（附下载链接）
前言自 2017 年 1 月 PyTorch 推出以来,其热度持续上升,一度有赶超 TensorFlow 的趋势.PyTorch 能在短时间内被众多研究人员和工程师接受并推崇是因为其有着诸多优点,如 ...
PyTorch 模型训练实用教程(附代码)
向AI转型的程序员都关注了这个号???????????? 机器学习AI算法工程公众号:datayx PyTorch 能在短时间内被众多研究人员和工程师接受并推崇是因为其有着诸多优点,如采用 Py ...
Pytorch - 弹性训练原理
Pytorch在1.9.0引入了torchrun,用其替代1.9.0以前版本的torch.distributed.launch.torchrun在torch.distributed.launch 功能 ...

最新文章

热门文章