针对零部件制造质量控制领域的缺陷检测问题,针对工业摄像头角度和零部件表面缺陷特征相对固定的特点,提出了基于注意力机制的YOLO缺陷检测算法的优化方案。该算法着重于提升算法的注意力,并采用了多种策略来实现。首先,采用了CZS算法,对图像中的缺陷区域进行了剪切、缩放和拼接,以便更好地集中于缺陷相关区域;其次,应用裁减主干网络算法,去除了原版YOLOv3主干网络中的无用检测尺度层;最后,利用数据增强算法增加了训练样本量。实验案例结果表明,该算法的检测精度达到了99.2%,单帧图像检测时间为0.01秒,性能优于原版YOLOv3;尤其在固定摄像头场景下,该算法具有一定的先进性,通过三项提升注意力的策略,算法训练精度更快地收敛,检测速度更快,检测性能更加稳定。
用户评论
推荐下载
-
注意力机制与Seq2seq模型.md
在seq2seq模型中,解码器只能隐式地从编码器的最终状态中选择相应的信息。然而,注意力机制可以将这种选择过程显式地建模。
10 2020-08-15 -
注意力驱动的网页设计
本资源仅包含网页分割数据,其他资源,包括代码,模型,数据,文章可进入网页自行下载(http://mmgroup.cs.cityu.edu.hk/projects/visualflow/),使用代码或者
12 2020-09-19 -
提升注意力的思维游戏
提升java程序员的逻辑思维,养成更好的良好的学习习惯!
40 2018-12-20 -
振纹检测缺陷检测
opencv3.4+vs2017 64位环境,测试图片一并打包,简单实现 工业零件振纹检测,振纹实际场景 情况比较多,这里只实现了 对颜色较深振纹的检测,通过 傅里叶变换,频率域滤波以及 形态学图像分
17 2021-05-05 -
一种基于注意力模型的面部表情识别算法
深度卷积网络提取的表情特征易受背景、个体身份等因素影响,其与无用特征混合在一起对表情识别造成干扰。针对此问题,提出一种基于注意力模型的面部表情识别算法,该方法基于一个轻量级的卷积神经网络以避免过拟合,
6 2021-02-01 -
基于金字塔注意力机制和GAN的水下图像增强
为提升水下图像的增强效果,提出了一种基于金字塔注意力机制和生成对抗网络(GAN)的水下图像增强算法。它将生成对抗网络作为基本架构,生成网络采用编码解码结构并引入特征金字塔注意力模块,多尺度金字塔特征与
15 2021-02-23 -
基于空时特征和注意力机制的无参考视频质量评价
随着视频技术的飞速发展,越来越多的视频应用逐步进入人们的生活中,因此对视频质量的研究很有意义。基于卷积神经网络和循环神经网络强大的特征提取能力并结合注意力机制,提出一种无参考视频质量评价算法。该算法首
8 2021-01-31 -
基于注意力机制和卷积神经网络的低照度图像增强
为了提高低照度图像的清晰度和避免颜色失真,提出了基于注意力机制和卷积神经网络(CNN)的低照度图像增强算法,以改善图像质量。首先根据Retinex模型合成训练数据,将原始图像从RGB (red-gre
27 2021-01-31 -
基于注意力机制的手写体中文字符识别
手写体中文的自动识别在文档数字化、手写笔记转录等方面有广泛应用。针对其具有的书写随意、结构复杂、数目众多等特点,提出了一种基于注意力机制的手写体中文识别方法。在卷积神经网络(CNN)模型的基础上,搭建
18 2021-01-31 -
ResNeSt注意力分散网络
It is well known that featuremap attention and multi-path representation are important for visual re
29 2021-01-22
暂无评论