前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >yolo3 检测出图像中的不规则汉字

yolo3 检测出图像中的不规则汉字

作者头像
机器学习AI算法工程
发布2019-10-28 17:17:22
1.4K0
发布2019-10-28 17:17:22
举报
1. 准备环境 (1)windows7 + python环境,安装anaconda3,我装的是python3.6版本 (2)安装tensorflow(cpu版)、keras,直接用pip install 命令下载,或conda install命令 (3)下载keras-yolo、下载VOC2007数据集;保留VOC2007中的每一个文件夹,但是要将文件夹中的所有小文件都删除 可以直接下载源码,里面已经配置好了VOC2007数据集。

完整代码获取方式:

关注微信公众号 datayx 然后回复 目标检测 即可获取。

2. 准备数据

1、准备了151张含有汉字的图像,将这些图像放在keras-yolo3-master\VOCdevkit\VOC2007\JPEGImages

2、用LabelImgv1.3.3标记这些图像,标记命名为"chinese",自动生成.xml文件,将这些.xml文件放在keras-yolo3-master\VOCdevkit\VOC2007\Annotations

3、生成ImageSets\Main\下的4个文件。运行VOCdevkit\VOC2007\下的test.py,会自动建好这四个文件。

4、生成keras-yolo3-master\下的test.txt、train.txt、val.txt,运行keras-yolo3-master\voc_annotation.py,会自动生成2007_test.txt、2007_train.txt、2007_val.txt三个文件,接下来需要我们手动去掉每一个文件名的前缀"2007_" test.py中的内容可参考: https://blog.csdn.net/Patrick_Lxc/article/details/80615433

3. 开始训练

1、修改参数配置文件yolo3.cfg,打开yolo3.cfg,搜索"yolo"关键词,能搜到三处包含"yolo"的地方,修改filters、classes、random classes = 1 # 因为我只检测汉字一类,所以是1 filters = 3 * (5+classe) = 3 * (5 + 1) = 18 random = 0

2、修改keras-yolo3-master\model_data下coco_classes.txt和voc_classes.txt,里面写的是要识别的类型,我写的是chinese。

3、运行keras-yolo3-master\train.py进行训练, 几个重要的参数如下: batch_size = 10 val_split = 0.1 epochs = 100

我使用CPU进行训练,程序跑了一天一夜【哭】,最后的loss在45.5左右,可以看keras-yolo3-master\训练日志.txt

4. 进行预测 运行keras-yolo3-master\yolo.py,识别的结果会存储到keras-yolo3-master\VOCdevkit\VOC2007\SegmentationClass中,部分结果如下:

via https://blog.csdn.net/u010429424/article/details/83238988

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2019-01-03,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 机器学习AI算法工程 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
相关产品与服务
图像识别
腾讯云图像识别基于深度学习等人工智能技术,提供车辆,物体及场景等检测和识别服务, 已上线产品子功能包含车辆识别,商品识别,宠物识别,文件封识别等,更多功能接口敬请期待。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档