advancedEAST算法

huoyijie大神(https://github.com/huoyijie/AdvancedEAST)改进了旷世提出的EAST算法,在长文本上效果良好。
advancedEAST网络结构图如下:
advancedEAST算法_第1张图片
采用vgg作为主网络,在特征提取层中增加了后面卷积层的通道数量,对后处理方法也进行了优化。
原作者自己也写了检测原理,我就不copy过来了:https://huoyijie.github.io/zh-Hans/2018/08/24/AdvancedEAST%E6%96%87%E6%9C%AC%E6%A3%80%E6%B5%8B%E5%8E%9F%E7%90%86%E7%AE%80%E4%BB%8B/
这是后置处理原理:https://huoyijie.github.io/zh-Hans/2018/08/27/AdvancedEAST%E5%90%8E%E7%BD%AE%E5%A4%84%E7%90%86%E5%8E%9F%E7%90%86%E7%AE%80%E4%BB%8B/

想去看看效果,所以下载了下代码
1下载预训练模型下载链接: https://pan.baidu.com/s/1KO7tR_MW767ggmbTjIJpuQ 提取码: kpm2
2创建文件夹saved_model,将下载后的模型文件解压后放到里面。
修改cfg.py文件里面的train_task_id,将该id修改与下载的预训练模型一致

train_task_id = ‘3T736’

3终端cd到目录后运行python predict.py -p demo/007.png
测了作者提供的图片
advancedEAST算法_第2张图片
advancedEAST算法_第3张图片

你可能感兴趣的:(advancedEAST算法)