CN112784748A - 一种基于改进的YOLOv3的微藻识别方法 - Google Patents
一种基于改进的YOLOv3的微藻识别方法 Download PDFInfo
- Publication number
- CN112784748A CN112784748A CN202110089255.XA CN202110089255A CN112784748A CN 112784748 A CN112784748 A CN 112784748A CN 202110089255 A CN202110089255 A CN 202110089255A CN 112784748 A CN112784748 A CN 112784748A
- Authority
- CN
- China
- Prior art keywords
- microalgae
- convolution
- data set
- improved
- training
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Pending
Links
- 238000000034 method Methods 0.000 title claims abstract description 37
- 238000001514 detection method Methods 0.000 claims abstract description 54
- 238000012549 training Methods 0.000 claims abstract description 27
- 238000000605 extraction Methods 0.000 claims abstract description 11
- 238000012360 testing method Methods 0.000 claims abstract description 11
- 238000012795 verification Methods 0.000 claims abstract description 7
- 238000002372 labelling Methods 0.000 claims abstract description 5
- 230000006870 function Effects 0.000 claims description 20
- 238000004364 calculation method Methods 0.000 claims description 17
- 238000010586 diagram Methods 0.000 claims description 12
- 238000011176 pooling Methods 0.000 claims description 11
- 238000012545 processing Methods 0.000 claims description 7
- 230000008569 process Effects 0.000 claims description 6
- 238000006243 chemical reaction Methods 0.000 claims description 3
- 238000000354 decomposition reaction Methods 0.000 claims description 3
- 238000001914 filtration Methods 0.000 claims description 3
- 230000001629 suppression Effects 0.000 claims description 3
- 238000005457 optimization Methods 0.000 abstract description 3
- 239000002245 particle Substances 0.000 abstract description 2
- XLYOFNOQVPJJNP-UHFFFAOYSA-N water Substances O XLYOFNOQVPJJNP-UHFFFAOYSA-N 0.000 description 11
- 239000000284 extract Substances 0.000 description 3
- 230000009545 invasion Effects 0.000 description 3
- 241000894007 species Species 0.000 description 3
- 241000195493 Cryptophyta Species 0.000 description 2
- 238000013135 deep learning Methods 0.000 description 2
- 238000013461 design Methods 0.000 description 2
- 238000011161 development Methods 0.000 description 2
- 239000013049 sediment Substances 0.000 description 2
- 238000012546 transfer Methods 0.000 description 2
- 102000004127 Cytokines Human genes 0.000 description 1
- 108090000695 Cytokines Proteins 0.000 description 1
- 230000004913 activation Effects 0.000 description 1
- 238000007796 conventional method Methods 0.000 description 1
- 238000013527 convolutional neural network Methods 0.000 description 1
- 230000007547 defect Effects 0.000 description 1
- 230000000694 effects Effects 0.000 description 1
- 230000007613 environmental effect Effects 0.000 description 1
- 230000004927 fusion Effects 0.000 description 1
- 238000010801 machine learning Methods 0.000 description 1
- 239000000463 material Substances 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
- 238000010606 normalization Methods 0.000 description 1
- 230000008520 organization Effects 0.000 description 1
- 244000052769 pathogen Species 0.000 description 1
- 238000011160 research Methods 0.000 description 1
- 238000006467 substitution reaction Methods 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V20/00—Scenes; Scene-specific elements
- G06V20/60—Type of objects
- G06V20/69—Microscopic objects, e.g. biological cells or cellular parts
- G06V20/693—Acquisition
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/21—Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
- G06F18/214—Generating training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/24—Classification techniques
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- Data Mining & Analysis (AREA)
- Life Sciences & Earth Sciences (AREA)
- General Physics & Mathematics (AREA)
- Artificial Intelligence (AREA)
- General Engineering & Computer Science (AREA)
- Evolutionary Computation (AREA)
- Computer Vision & Pattern Recognition (AREA)
- General Health & Medical Sciences (AREA)
- Bioinformatics & Computational Biology (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Health & Medical Sciences (AREA)
- Biomedical Technology (AREA)
- Molecular Biology (AREA)
- Evolutionary Biology (AREA)
- Computational Linguistics (AREA)
- Biophysics (AREA)
- Computing Systems (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Multimedia (AREA)
- Image Analysis (AREA)
Abstract
本发明提供一种基于改进的YOLOv3的微藻识别方法,包括:采集微藻显微图像,制作微藻图像的数据集;对数据集进行数据增强;将增强后的数据集分为训练集、验证集和测试集,对数据集中的微藻进行标注,生成标注图像;构建改进的YOLOv3目标检测模型;设置训练参数,基于数据集对构建的YOLOv3目标检测模型进行训练;基于训练完成的YOLOv3目标检测模型,对测试集图像进行分类和定位。本发明采用改进的YOLOv3目标检测模型,使用轻量级Mobilenet网络代替YOLOv3的原始特征提取网络darknet53,能够显著提高运行速度,大大减少网络参数,同时引入空间金字塔池结构SPP,能够在同一卷积层中以不同尺度合并和连接区域特征,使得在检测小物体时位置误差较小,使用CIoU优化损耗功能进一步提高检测精度。
Description
技术领域
本发明涉及海洋及船舶压载水中微藻种类识别技术领域,具体而言,尤其涉及一种基于改进的YOLOv3的微藻识别方法。
背景技术
船舶压载水对船舶在航行中保持安全稳定的状态具有重要作用,但国际航行船舶排放的压载水为外来生物入侵提供了渠道。随着经济全球化的发展,海运贸易量快速增长,船舶压载水造成的生态入侵问题逐步显现,引起了全球环保和航运领域广泛关注,被公认为全球海洋生态和海洋经济的重大威胁之一。国际海洋考察理事会(ICES)的数据表明94%的潜在有害海洋生物是通过船舶压载水传播。为防止、尽量减少和最终消除船舶压载水和沉积物造成的外来水生物和病原体的转移,保护海洋生态环境,国际海事组织(IMO)于2004年制定了《国际船舶压载水及其沉积物控制与管理公约》(以下简称《公约》),该公约于2017年9月8日正式生效。当前,在压载水入港的病媒检测中,微藻占据主要角色,因此对压载水中微藻进行检测是迫在眉睫的任务。藻类的种类和数量是压载水检测中的关键指标,目前常规方法是人工检测,这种方法存在很大的缺陷:检测设备步骤繁琐、体积大、价格昂贵,无法满足市场的要求,且人工检测中对工作人员的技能和熟练度要求很高,而且在大量检测过程中,工作人员容易疲劳和缺乏客观性,使得检测结果不合格、错误率高。藻类分类识别一直是生物学领域的一个热点问题,现在有很多使用传统识别算法替代人工检测,传统的机器学习识别方法通常采用特征提取与分类器相结合的方式,但是需要人为选择特征参数,存在对手工设计特征依赖的问题,影响识别结果的准确性和稳定性。随着卷积神经网络的迅速发展,利用深度学习可以进行微藻的识别。目标检测领域的深度学习算法主要分为两阶段目标检测算法和单阶段目标检测算法,其中两阶段目标检测算法存在着耗费检测时间的问题;单阶段目标检测算法的检测速度快,但识别具有相似特征的不同目标准确率低,尤其是在当前的目标检测领域,大部分都是用于宏观物体上的检测,很少用于微观领域中,因此需要考虑当对微观物体进行检测时,是否会出现一些新的问题。
发明内容
根据上述提出的技术问题,而提供一种基于改进的YOLOv3的微藻识别方法。本发明采用改进的YOLOv3目标检测模型,使用轻量级Mobilenet网络代替YOLOv3的原始特征提取网络darknet53,能够显著提高运行速度,大大减少网络参数,同时引入空间金字塔池结构SPP,能够在同一卷积层中以不同尺度合并和连接区域特征,使得在检测小物体时位置误差较小,使用CIoU优化损耗功能进一步提高检测精度。
本发明采用的技术手段如下:
一种基于改进的YOLOv3的微藻识别方法,包括如下步骤:
S1、采集微藻显微图像,制作微藻图像的数据集;
S2、对所述数据集进行数据增强;
S3、将增强后的数据集分为训练集、验证集和测试集,并对数据集中的微藻进行标注,生成XML格式的标注图像;
S4、构建改进的YOLOv3目标检测模型;
S5、设置训练参数,基于数据集对构建的YOLOv3目标检测模型进行训练;
S6、基于训练完成的YOLOv3目标检测模型,对测试集图像进行分类和定位。
进一步地,所述步骤S1具体为:
通过Ts2-FL/TS2荧光显微镜将每种微藻在不同放大倍数及亮度下进行拍摄,且每种微藻图像数量均等。
进一步地,所述步骤S2具体为:
采用剪裁、平移、改变亮度、旋转角度以及镜像的变换方法,将数据集图像缩放20%并进行局部模糊处理。
进一步地,所述步骤S3具体为:
将增强后的数据集按照8:1:1的比例,且每种微藻图像数量以均等的方式分为训练集、验证集和测试集。
进一步地,所述步骤S4具体包括:
S41、采用轻量级Mobilenet网络代替YOLOv3的原始特征提取darknet53网络;
S42、引入空间金字塔池结构SPP模块,在同一卷积层中以不同尺度合并和连接区域特征;
S43、采用CIoU函数作为微藻识别的分类损失函数优化损耗功能。
进一步地,所述步骤S41具体为:
S411、MobileNet网络将常规卷积分解为深度卷积和逐点卷积;
S412、深度卷积将标准卷积分为两个单独的层,用于过滤和合并;
S413、逐点卷积通过使用1×1卷积来组合输出深度卷积;
S414、基于上述分解,深度可分离卷积计算成本为:
C1=DK×DK×M×DF×DF+M×N×DF×DF
上式中,M表示输入通道数,N表示输出特征图的通道数,DK表示卷积运算内核的长和宽,DF表示输入与输出特征图的宽度与高度;
S415、在输入与输出图像尺寸一致的情况下,传统卷积的计算成本为:
C2=DK×DK×M×N×DF×DF
S416、将深度可分离卷积计算成本和传统卷积的计算成本的计算量相对比,得到深度可分离卷积与传统卷积计算量的比值为:
进一步地,所述步骤S42具体为:
将空间金字塔池结构SPP掺杂在特征网络最后一个特征层的卷积里,在特征网络的最后一个特征层进行三次卷积后,分别利用四个不同尺度的最大池化进行处理,得到同样长度的池化特征。
进一步地,所述步骤S43中的CIoU函数,其计算公式如下:
上式中,IoU表示预测边界框和目标边界框的交集比率,A表示预测边界框,B表示目标边界框;b和bgt表示A和B的中心点,ρ表示b和bgt的欧式距离,c表示可以覆盖A和B的最小矩形的对角线距离,α表示用于作trade-off的参数,为权重函数,v表示用来衡量长宽比一致性的参数,
进一步地,所述步骤S5具体包括:
S51、使用在ImageNet数据集上的预训练权重模型作为改进的YOLOv3初始化权重模型;
S52、训练过程中,采用随机梯度下降算法优化训练模型,初始学习率为0.001,权重衰减为0.0001,动量因子为0.9,epoch第一阶段初始化为0-70,迭代到第45个epoch,学习率降为上阶段的10%,迭代到第64个epoch,学习率再降为上阶段的10%;epoch第二阶段初始化为70-150,迭代到第75个epoch,学习率降为上阶段的10%,迭代到第119个epoch,学习率降为上阶段的10%,每批次训练数目为8;
S53、当锚框anchor与真实标记框的交并比IoU大于0.5时,anchor标记为正样本,当anchor与真实标记框的IoU小于0.35时,anchor标记为负样本,当anchor与真实标记框的IoU在0.3和0.5之间时,在训练期间忽略该anchor,非极大值抑制设置为0.6,去除冗余的检测框,找到物体检测位置。
较现有技术相比,本发明具有以下优点:
1、本发明提供的基于改进的YOLOv3的微藻识别方法,不需要耗费大量的人力物力,解决对手工设计特征依赖的问题,图像采集方法操作简单。
2、本发明提供的基于改进的YOLOv3的微藻识别方法,采用改进的YOLOv3目标检测模型,使用轻量级Mobilenet网络代替YOLOv3的原始特征提取网络darknet53,能够显著提高运行速度,同时大大减少网络参数。
3、本发明提供的基于改进的YOLOv3的微藻识别方法,引入一种空间金字塔池结构SPP,能够在同一卷积层中以不同尺度合并和连接区域特征,使得在检测小物体时位置误差较小。
4、本发明提供的基于改进的YOLOv3的微藻识别方法,使用CIoU优化损耗功能进一步提高检测精度。
5、本发明提供的基于改进的YOLOv3的微藻识别方法,解决了识别具有相似特征的不同目标准确率低的问题,且目标检测用于微观领域中,提高了对小目标的检测精度。
基于上述理由本发明可在海洋及船舶压载水中微藻种类识别等领域广泛推广。
附图说明
为了更清楚地说明本发明实施例或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图做以简单地介绍,显而易见地,下面描述中的附图是本发明的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动性的前提下,还可以根据这些附图获得其他的附图。
图1为本发明方法流程图。
图2为本发明实施例提供的拍摄装置示意图。
图3为本发明实施例提供的改进的YOLOv3目标检测模型示意图。
图4为本发明实施例提供的轻量级MobileNet网络示意图。
图5为本发明实施例提供的空间金字塔池结构SPP模块示意图。
具体实施方式
为了使本技术领域的人员更好地理解本发明方案,下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本发明一部分的实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都应当属于本发明保护的范围。
需要说明的是,本发明的说明书和权利要求书及上述附图中的术语“第一”、“第二”等是用于区别类似的对象,而不必用于描述特定的顺序或先后次序。应该理解这样使用的数据在适当情况下可以互换,以便这里描述的本发明的实施例能够以除了在这里图示或描述的那些以外的顺序实施。此外,术语“包括”和“具有”以及他们的任何变形,意图在于覆盖不排他的包含,例如,包含了一系列步骤或单元的过程、方法、系统、产品或设备不必限于清楚地列出的那些步骤或单元,而是可包括没有清楚地列出的或对于这些过程、方法、产品或设备固有的其它步骤或单元。
如图1所示,本发明提供了一种基于改进的YOLOv3的微藻识别方法,包括如下步骤:
S1、采集微藻显微图像,制作微藻图像的数据集;
具体实施时,作为本发明优选的实施方式,所述步骤S1具体为:
采集的微藻显微图像为实验室显微图像,如图2所示,通过Ts2-FL/TS2荧光显微镜将每种微藻在不同放大倍数及亮度下进行拍摄,且每种微藻图像数量均等。
S2、对所述数据集进行数据增强;
具体实施时,作为本发明优选的实施方式,所述步骤S2具体为:
采用剪裁、平移、改变亮度、旋转角度以及镜像的变换方法,将数据集图像缩放20%并进行局部模糊处理。
S3、将增强后的数据集分为训练集、验证集和测试集,通过labelImg图像标注工具并对数据集中的微藻进行标注,生成XML格式的标注图像;
具体实施时,作为本发明优选的实施方式,所述步骤S3具体为:
将增强后的数据集按照8:1:1的比例,且每种微藻图像数量以均等的方式分为训练集、验证集和测试集。
S4、构建改进的YOLOv3目标检测模型;本实施例中,YOLOv3设计了目标多尺度检测结构,并使用了logistic功能代替了传统的softmax功能。YOLOv3将最后三个下采样层传输到检测层以进行目标检测,YOLO-V3的网络可以从三个方面进行预测,这三个标尺的大小分别为13×13、26×26和52×52,分别用于检测大目标,中目标和小目标。采用多尺度检测层(三尺度)来检测目标,使YOLO-V3可以更有效地检测小目标。在Darknet-53中,尽管引入了许多1×1卷积核,并且使用步长为2的3×3卷积核代替了最大池,但参数数量却大大减少了。但是,Darknet-53网络仍然有些复杂和冗余。为了减少YOLOv3模型中的参数数量并提高其检测速度,对其进行了改进,如图3所示,为本发明实施例的改进的YOLOv3目标检测网络模型示意图。
具体实施时,作为本发明优选的实施方式,所述步骤S4具体包括:
S41、采用轻量级Mobilenet网络代替YOLOv3的原始特征提取darknet53网络;如图4所示,为本发明实施例的轻量级MobileNet网络示意图。
所述步骤S41具体为:
S411、MobileNet网络将常规卷积分解为深度卷积和逐点卷积;
S412、深度卷积将标准卷积分为两个单独的层,用于过滤和合并;
S413、逐点卷积通过使用1×1卷积来组合输出深度卷积;
本实施例中,将MobileNet网络分割成三块不同特征提取图,第一个提取层提取出52×52×256特征图用于检测较小物体,第二个提取层提取出26×26×512特征图用于检测中等物体,第三个提取层提取出13×13×1024特征图用于检测较大物体。在深度可分离卷积计算成本中,减少了卷积和参数的数量。这种方式大大减少了计算量和模型大小,减少了冗余,保证精度的同时又提高了网络的实时性。
S414、基于上述分解,深度可分离卷积计算成本为:
C1=DK×DK×M×DF×DF+M×N×DF×DF
上式中,M表示输入通道数,N表示输出特征图的通道数,DK表示卷积运算内核的长和宽,DF表示输入与输出特征图的宽度与高度;
S415、在输入与输出图像尺寸一致的情况下,传统卷积的计算成本为:
C2=DK×DK×M×N×DF×DF
S416、将深度可分离卷积计算成本和传统卷积的计算成本的计算量相对比,得到深度可分离卷积与传统卷积计算量的比值为:
从上式可得,深度可分离卷积与传统卷积计算量的比值为MobileNet作为基础网络,提取图片特征,可以在尽可能保证性能的情况下,极大地减少运算量。深度可分离卷积在每次卷积之后都应用了批量标准化(BN)和ReLU块。MobileNet采用ReLU作为激活函数,使用更多的ReLU,既增加了模型的非线性变化,又提高了模型的泛化能力。
S42、引入空间金字塔池结构SPP模块,在同一卷积层中以不同尺度合并和连接区域特征;如图5所示,为本发明实施例的空间金字塔池结构SPP模块示意图。SPP-net全名为空间金字塔池化结构,是计算机视觉最流行的方法之一,SPP-net采用多级窗口对输入的特征图进行池化操作,对于给定尺寸的输入图像,可计算出SPP-net的池化窗口参数,假设输入的特征图大小为a×b(a和b的大小是可变的),对于一个包含n×n个空间盒(单位为bins)的金字塔层,其池化窗口大小为w1×w2,w1、w2分别为窗口的宽和高,对应步长为:t1、t2分别表示水平和竖直步长,其中和分别表示向上和向下取整。对于有l层的金字塔,每层都按此规则进行池化操作,最终将各金字塔层级池化结果进行合并就会得到固定长度的输出,作为全连接层的输入。
YOLOv3的多尺度预测侧重于级联多尺度卷积层的全局特征,而忽略了同一卷积层上多尺度局部区域特征的融合。而且,尽管YOLOv3引入的FPN结构提高了小目标检测的准确性,但对中型和大型对象的检测性能却相对较差。因此,具体实施时,作为本发明优选的实施方式,所述步骤S42具体为:
将空间金字塔池结构SPP掺杂在特征网络最后一个特征层的卷积里,在特征网络的最后一个特征层进行三次卷积后,分别利用四个不同尺度的最大池化进行处理,得到同样长度的池化特征。
SPP结构,受SPP-net的启发而来,SPP结构掺杂在特征网络的最后一个特征层的卷积里,在特征网络的最后一个特征层进行三次卷积后,分别利用四个不同尺度的最大池化进行处理,最大池化的池化核大小分别为13x13、9x9、5x5、1x1(1x1即无处理)。SPP显著特点为:(1)不管输入尺寸是怎样,SPP可以产生固定大小的输出,(2)使用多个窗口。(3)SPP可以使用同一图像不同尺寸作为输入,得到同样长度的池化特征。
S43、采用CIoU函数作为微藻识别的分类损失函数优化损耗功能。
具体实施时,作为本发明优选的实施方式,所述步骤S43中的CIoU函数,其计算公式如下:
上式中,IoU表示预测边界框和目标边界框的交集比率,A表示预测边界框,B表示目标边界框;b和bgt表示A和B的中心点,ρ表示b和bgt的欧式距离,c表示可以覆盖A和B的最小矩形的对角线距离,α表示用于作trade-off的参数,为权重函数,v表示用来衡量长宽比一致性的参数,
S5、设置训练参数,基于数据集对构建的YOLOv3目标检测模型进行训练;
具体实施时,作为本发明优选的实施方式,所述步骤S5具体包括:
S51、使用在ImageNet数据集上的预训练权重模型作为改进的YOLOv3初始化权重模型;
S52、训练过程中,采用随机梯度下降算法优化训练模型,初始学习率为0.001,权重衰减为0.0001,动量因子为0.9,epoch第一阶段初始化为0-70,迭代到第45个epoch,学习率降为上阶段的10%,迭代到第64个epoch,学习率再降为上阶段的10%;epoch第二阶段初始化为70-150,迭代到第75个epoch,学习率降为上阶段的10%,迭代到第119个epoch,学习率降为上阶段的10%,每批次训练数目为8;
S53、当锚框anchor与真实标记框的交并比IoU大于0.5时,anchor标记为正样本,当anchor与真实标记框的IoU小于0.35时,anchor标记为负样本,当anchor与真实标记框的IoU在0.3和0.5之间时,在训练期间忽略该anchor,非极大值抑制设置为0.6,去除冗余的检测框,找到物体检测位置。
S6、基于训练完成的YOLOv3目标检测模型,对测试集图像进行分类和定位。
最后应说明的是:以上各实施例仅用以说明本发明的技术方案,而非对其限制;尽管参照前述各实施例对本发明进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分或者全部技术特征进行等同替换;而这些修改或者替换,并不使相应技术方案的本质脱离本发明各实施例技术方案的范围。
Claims (9)
1.一种基于改进的YOLOv3的微藻识别方法,其特征在于,包括如下步骤:
S1、采集微藻显微图像,制作微藻图像的数据集;
S2、对所述数据集进行数据增强;
S3、将增强后的数据集分为训练集、验证集和测试集,并对数据集中的微藻进行标注,生成XML格式的标注图像;
S4、构建改进的YOLOv3目标检测模型;
S5、设置训练参数,基于数据集对构建的YOLOv3目标检测模型进行训练;
S6、基于训练完成的YOLOv3目标检测模型,对测试集图像进行分类和定位。
2.根据权利要求1所述的基于改进的YOLOv3的微藻识别方法,其特征在于,所述步骤S1具体为:
通过Ts2-FL/TS2荧光显微镜将每种微藻在不同放大倍数及亮度下进行拍摄,且每种微藻图像数量均等。
3.根据权利要求1所述的基于改进的YOLOv3的微藻识别方法,其特征在于,所述步骤S2具体为:
采用剪裁、平移、改变亮度、旋转角度以及镜像的变换方法,将数据集图像缩放20%并进行局部模糊处理。
4.根据权利要求1所述的基于改进的YOLOv3的微藻识别方法,其特征在于,所述步骤S3具体为:
将增强后的数据集按照8:1:1的比例,且每种微藻图像数量以均等的方式分为训练集、验证集和测试集。
5.根据权利要求1所述的基于改进的YOLOv3的微藻识别方法,其特征在于,所述步骤S4具体包括:
S41、采用轻量级Mobilenet网络代替YOLOv3的原始特征提取darknet53网络;
S42、引入空间金字塔池结构SPP模块,在同一卷积层中以不同尺度合并和连接区域特征;
S43、采用CIoU函数作为微藻识别的分类损失函数优化损耗功能。
6.根据权利要求5所述的基于改进的YOLOv3的微藻识别方法,其特征在于,所述步骤S41具体为:
S411、MobileNet网络将常规卷积分解为深度卷积和逐点卷积;
S412、深度卷积将标准卷积分为两个单独的层,用于过滤和合并;
S413、逐点卷积通过使用1×1卷积来组合输出深度卷积;
S414、基于上述分解,深度可分离卷积计算成本为:
C1=DK×DK×M×DF×DF+M×N×DF×DF
上式中,M表示输入通道数,N表示输出特征图的通道数,DK表示卷积运算内核的长和宽,DF表示输入与输出特征图的宽度与高度;
S415、在输入与输出图像尺寸一致的情况下,传统卷积的计算成本为:
C2=DK×DK×M×N×DF×DF
S416、将深度可分离卷积计算成本和传统卷积的计算成本的计算量相对比,得到深度可分离卷积与传统卷积计算量的比值为:
7.根据权利要求5所述的基于改进的YOLOv3的微藻识别方法,其特征在于,所述步骤S42具体为:
将空间金字塔池结构SPP掺杂在特征网络最后一个特征层的卷积里,在特征网络的最后一个特征层进行三次卷积后,分别利用四个不同尺度的最大池化进行处理,得到同样长度的池化特征。
9.根据权利要求1所述的基于改进的YOLOv3的微藻识别方法,其特征在于,所述步骤S5具体包括:
S51、使用在ImageNet数据集上的预训练权重模型作为改进的YOLOv3初始化权重模型;
S52、训练过程中,采用随机梯度下降算法优化训练模型,初始学习率为0.001,权重衰减为0.0001,动量因子为0.9,epoch第一阶段初始化为0-70,迭代到第45个epoch,学习率降为上阶段的10%,迭代到第64个epoch,学习率再降为上阶段的10%;epoch第二阶段初始化为70-150,迭代到第75个epoch,学习率降为上阶段的10%,迭代到第119个epoch,学习率降为上阶段的10%,每批次训练数目为8;
S53、当锚框anchor与真实标记框的交并比IoU大于0.5时,anchor标记为正样本,当anchor与真实标记框的IoU小于0.35时,anchor标记为负样本,当anchor与真实标记框的IoU在0.3和0.5之间时,在训练期间忽略该anchor,非极大值抑制设置为0.6,去除冗余的检测框,找到物体检测位置。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110089255.XA CN112784748A (zh) | 2021-01-22 | 2021-01-22 | 一种基于改进的YOLOv3的微藻识别方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110089255.XA CN112784748A (zh) | 2021-01-22 | 2021-01-22 | 一种基于改进的YOLOv3的微藻识别方法 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN112784748A true CN112784748A (zh) | 2021-05-11 |
Family
ID=75758649
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202110089255.XA Pending CN112784748A (zh) | 2021-01-22 | 2021-01-22 | 一种基于改进的YOLOv3的微藻识别方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN112784748A (zh) |
Cited By (7)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113011405A (zh) * | 2021-05-25 | 2021-06-22 | 南京柠瑛智能科技有限公司 | 一种解决无人机地物目标识别多框重叠误差的方法 |
CN113435269A (zh) * | 2021-06-10 | 2021-09-24 | 华东师范大学 | 一种基于YOLOv3改进的水面漂浮物检测与识别方法及系统 |
CN115019243A (zh) * | 2022-04-21 | 2022-09-06 | 山东大学 | 基于改进YOLOv3的监控漂浮物轻量化目标检测方法及系统 |
CN115187982A (zh) * | 2022-07-12 | 2022-10-14 | 河北华清环境科技集团股份有限公司 | 藻类检测方法、装置及终端设备 |
TWI786711B (zh) * | 2021-07-07 | 2022-12-11 | 台灣電力股份有限公司 | 智慧型微藻養殖系統及其方法 |
CN115690546A (zh) * | 2022-12-30 | 2023-02-03 | 正大农业科学研究有限公司 | 虾长测量方法、装置、电子设备及存储介质 |
CN115775381A (zh) * | 2022-12-15 | 2023-03-10 | 华洋通信科技股份有限公司 | 一种光照不均匀下的矿井电机车路况识别方法 |
Citations (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN109635875A (zh) * | 2018-12-19 | 2019-04-16 | 浙江大学滨海产业技术研究院 | 一种基于深度学习的端到端网口检测方法 |
CN109858386A (zh) * | 2019-01-09 | 2019-06-07 | 深圳市趣方科技有限公司 | 一种基于荧光显微图像的微藻细胞识别方法 |
CN110766098A (zh) * | 2019-11-07 | 2020-02-07 | 中国石油大学(华东) | 基于改进YOLOv3的交通场景小目标检测方法 |
CN112215071A (zh) * | 2020-09-10 | 2021-01-12 | 华蓝设计(集团)有限公司 | 异质交通流下自动驾驶车载多目标耦合识别与追踪方法 |
-
2021
- 2021-01-22 CN CN202110089255.XA patent/CN112784748A/zh active Pending
Patent Citations (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN109635875A (zh) * | 2018-12-19 | 2019-04-16 | 浙江大学滨海产业技术研究院 | 一种基于深度学习的端到端网口检测方法 |
CN109858386A (zh) * | 2019-01-09 | 2019-06-07 | 深圳市趣方科技有限公司 | 一种基于荧光显微图像的微藻细胞识别方法 |
CN110766098A (zh) * | 2019-11-07 | 2020-02-07 | 中国石油大学(华东) | 基于改进YOLOv3的交通场景小目标检测方法 |
CN112215071A (zh) * | 2020-09-10 | 2021-01-12 | 华蓝设计(集团)有限公司 | 异质交通流下自动驾驶车载多目标耦合识别与追踪方法 |
Non-Patent Citations (6)
Title |
---|
KAIMING HE ETC: "Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition", 《IEEE》, pages 2 * |
徐 镪等: "改进的YOLOv3网络在钢板表面缺陷检测研究", 《计算机工程与应用》, pages 265 - 271 * |
李锦泽: "基于深度学习的行人检测技术研究", 《中国优秀硕士学位论文全文数据库(电子期刊)》, pages 4 * |
王晓阳: "基于图像的海洋微藻识别系统的设计与实现", 《中国优秀硕士学位论文全文数据库(电子期刊)》, pages 4 * |
邵伟平;王兴;曹昭睿;白帆;: "基于MobileNet与YOLOv3的轻量化卷积神经网络设计", 计算机应用, no. 1 * |
邵伟平;王兴;曹昭睿;白帆;: "基于MobileNet与YOLOv3的轻量化卷积神经网络设计", 计算机应用, no. 1, 10 July 2020 (2020-07-10) * |
Cited By (10)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113011405A (zh) * | 2021-05-25 | 2021-06-22 | 南京柠瑛智能科技有限公司 | 一种解决无人机地物目标识别多框重叠误差的方法 |
CN113011405B (zh) * | 2021-05-25 | 2021-08-13 | 南京柠瑛智能科技有限公司 | 一种解决无人机地物目标识别多框重叠误差的方法 |
CN113435269A (zh) * | 2021-06-10 | 2021-09-24 | 华东师范大学 | 一种基于YOLOv3改进的水面漂浮物检测与识别方法及系统 |
TWI786711B (zh) * | 2021-07-07 | 2022-12-11 | 台灣電力股份有限公司 | 智慧型微藻養殖系統及其方法 |
CN115019243A (zh) * | 2022-04-21 | 2022-09-06 | 山东大学 | 基于改进YOLOv3的监控漂浮物轻量化目标检测方法及系统 |
CN115187982A (zh) * | 2022-07-12 | 2022-10-14 | 河北华清环境科技集团股份有限公司 | 藻类检测方法、装置及终端设备 |
CN115775381A (zh) * | 2022-12-15 | 2023-03-10 | 华洋通信科技股份有限公司 | 一种光照不均匀下的矿井电机车路况识别方法 |
CN115775381B (zh) * | 2022-12-15 | 2023-10-20 | 华洋通信科技股份有限公司 | 一种光照不均匀下的矿井电机车路况识别方法 |
CN115690546A (zh) * | 2022-12-30 | 2023-02-03 | 正大农业科学研究有限公司 | 虾长测量方法、装置、电子设备及存储介质 |
CN115690546B (zh) * | 2022-12-30 | 2023-05-05 | 正大农业科学研究有限公司 | 虾长测量方法、装置、电子设备及存储介质 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN112784748A (zh) | 一种基于改进的YOLOv3的微藻识别方法 | |
CN112580439A (zh) | 小样本条件下的大幅面遥感图像舰船目标检测方法及系统 | |
WO2020046213A1 (en) | A method and apparatus for training a neural network to identify cracks | |
CN110060508B (zh) | 一种用于内河桥区的船舶自动检测方法 | |
CN112906485A (zh) | 基于改进的yolo模型的视障人士辅助障碍物感知方法 | |
CN114842208A (zh) | 一种基于深度学习的电网危害鸟种目标检测方法 | |
CN112488020A (zh) | 基于无人机航拍数据的水环境污染情况检测评估装置 | |
Li et al. | A robust real‐time method for identifying hydraulic tunnel structural defects using deep learning and computer vision | |
CN114155474A (zh) | 基于视频语义分割算法的损伤识别技术 | |
Cao et al. | Detection of microalgae objects based on the Improved YOLOv3 model | |
CN116229248A (zh) | 海洋物种的分布预测方法、装置、设备及存储介质 | |
CN114005064A (zh) | 一种基于机器视觉技术的生物式水体污染预警方法及装置 | |
CN114170422A (zh) | 一种煤矿井下图像语义分割方法 | |
CN113469097A (zh) | 一种基于ssd网络的水面漂浮物多相机实时检测方法 | |
Shuai et al. | Regression convolutional network for vanishing point detection | |
CN111724338A (zh) | 一种转盘异常识别方法、系统、电子设备及介质 | |
Wu et al. | Research on asphalt pavement disease detection based on improved YOLOv5s | |
CN114882375A (zh) | 一种尾矿库智能识别方法和装置 | |
CN114663683A (zh) | 一种基于空间特征自监督的水下目标检测方法 | |
Wang et al. | Waste-YOLO: towards high accuracy real-time abnormal waste detection in waste-to-energy power plant for production safety | |
Li | Construction method of swimming pool intelligent assisted drowning detection model based on computer feature pyramid networks | |
CN113505699A (zh) | 一种基于RetinaNet算法的船舶检测方法 | |
CN113591773A (zh) | 基于卷积神经网络的配电室物体检测方法、装置和设备 | |
Kumar et al. | Real Time Prediction Model for Air Pollution and Air Quality Index based on Machine Learning | |
CN111724337B (zh) | 一种冷床冲顶识别方法、系统、电子设备及介质 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination |