search

yizt / crnn.pytorch

crnn实现水平和垂直方向中文文字识别, 提供在3w多个中文字符训练的水平识别和垂直识别的预训练模型; 欢迎关注,试用和反馈问题... ...
Apache License 2.0
242 stars 52 forks source link

请教一下如何训练自己的数据集，数据格式以及存放目录怎么配置 #13

Open zltgo opened 4 years ago

zltgo commented 4 years ago

如题

yizt commented 4 years ago

@zltgo 您好，对于自己的数据，没有任何要求，根据自己的数据写一个torch.utils.data.dataset.Dataset就好，就像Generator类

zltgo commented 4 years ago

thank you!