CRAFT(Character Region Awareness for Text Detection)的训练方法
Posted SpikeKing
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了CRAFT(Character Region Awareness for Text Detection)的训练方法相关的知识，希望对你有一定的参考价值。
- 工程：CRAFT-Reimplementation
- 数据集：Syndata - SynthText，数据集比较大，大约40G
- PyTorch版本：Which version of python and CUDA is used ?
下载PyTorch的CUDA10版本
conda install pytorch==1.0.1 torchvision==0.2.2 cudatoolkit=10.0
PyTorch的历史版本，找到cu100版本，下载torch-1.0.1-cp37-cp37m-linux_x86_64.whl，torchvision正在pip安装即可。
修改对应的数据和模型路径
即可训练。
以上是关于CRAFT(Character Region Awareness for Text Detection)的训练方法的主要内容，如果未能解决你的问题，请参考以下文章
如何将工艺（用于文本检测的字符区域感知）的输出提供给 crnn
十分钟了解兵器榜排行第一的 Sketch 插件 ：CRAFT