一种基于深度学习对视频或图片中水线的检测和识别方法

基本信息

申请号 CN201810290723.8 申请日 -
公开(公告)号 CN108509919A 公开(公告)日 2018-09-07
申请公布号 CN108509919A 申请公布日 2018-09-07
分类号 G06K9/00;G06N3/04 分类 计算;推算;计数;
发明人 朱齐丹;吕晓龙;游锦成;吴禹辰 申请(专利权)人 哈尔滨哈船智控科技有限责任公司
代理机构 哈尔滨市阳光惠远知识产权代理有限公司 代理人 哈尔滨哈船智控科技有限责任公司
地址 150001 黑龙江省哈尔滨市南岗区南通大街258号哈尔滨船舶电子大世界西区5层第501-1号
法律状态 -

摘要

摘要 本发明提出了一种基于深度学习对视频或图片中水线的检测和识别方法,采集训练数据并制作训练样本数据集,采用faster‑rcnn算法,联合训练rpn网络和识别网络,再由训练完成后的模型对实时采集的视频或图片进行识别,由算法给出的水线在图像中的多数位置信息,再对少量误识别的水线进行去除,并对漏识别的水线区域进行预测,增加了本发明在复杂条件下的应用能力。本发明在各种天气、光照、水质等条件下取得丰富的训练样本,并对图像中的水线的标记进行合理地设计,极大减少了误识别或不识别的情况,并有很快的检测速度。