CN113343949A - 一种通用的嵌入式平台的行人检测模型训练方法 - Google Patents
一种通用的嵌入式平台的行人检测模型训练方法 Download PDFInfo
- Publication number
- CN113343949A CN113343949A CN202110883375.7A CN202110883375A CN113343949A CN 113343949 A CN113343949 A CN 113343949A CN 202110883375 A CN202110883375 A CN 202110883375A CN 113343949 A CN113343949 A CN 113343949A
- Authority
- CN
- China
- Prior art keywords
- model
- training
- pedestrian detection
- fine
- detection model
- 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.)
- Granted
Links
Images
Classifications
-
- 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/21—Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
- G06F18/217—Validation; Performance evaluation; Active pattern learning 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/08—Learning methods
- G06N3/084—Backpropagation, e.g. using gradient descent
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T1/00—General purpose image data processing
- G06T1/20—Processor architectures; Processor configuration, e.g. pipelining
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- Data Mining & Analysis (AREA)
- General Physics & Mathematics (AREA)
- Life Sciences & Earth Sciences (AREA)
- Artificial Intelligence (AREA)
- General Engineering & Computer Science (AREA)
- Evolutionary Computation (AREA)
- Bioinformatics & Computational Biology (AREA)
- Evolutionary Biology (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Computational Linguistics (AREA)
- Biomedical Technology (AREA)
- Biophysics (AREA)
- Health & Medical Sciences (AREA)
- General Health & Medical Sciences (AREA)
- Molecular Biology (AREA)
- Computing Systems (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Image Analysis (AREA)
Abstract
一种通用的基于嵌入式平台的行人检测模型的训练方法,包括:数据集的准备,进行图像标准化;训练浮点模型,采用降低浮点数进行深度学习模型训练;微调量化卷积层,使用上一步骤得到的浮点模型,并将模型的路径添加到微调量化卷积脚本中进行微调,得到微调量化卷积层;应用线性变换微调模型,利用已标注的样本对预训练参数进行调整,以针对行人检测任务来修正模型网络;模型效果测试,利用测试集测试行人检测模型的准确率和召回率;在实时性和检测的灵活性上有较大的提高,同时具有高准确率和召回率等良好性能。
Description
技术领域
本发明属于人工智能领域,具体涉及一种将计算机图像视觉技术中的行人检测应用于嵌入式平台的方法。
背景技术
目标检测是当前计算机视觉所研究的重要领域之一,其目的是能够对图像或视频中的特定目标进行定位,并在原始图像中标定出目标的具体位置。行人检测是其中应用范围很广的一个具体实例,在许多重要场合都有非常重要的应用。传统的行人检测方法是通过设计和提取人工设计的分辨率较强的特征,如梯度直方图特征,Haar特征等,接着使用支持向量机等机器学习中的分类模型对所提取得到的特征进行分类,从而分别出是否是行人目标。然而,在实际场景下,行人容易受到光照、姿势、比例和拍摄角度等诸多因素的限制和影响,这些存在的因素严重影响了使用传统方法进行行人检测的准确度。自从深度学习被广泛应用于各种计算机视觉任务以来,各个领域都得到了飞速的发展。在目标检测领域,深度学习的出现大幅度提高了检测的准确度和鲁棒性,但是卷积神经网络所存在的最大的问题是模型参数量大,预测时计算量大使得模型的实时性并不高,MobileNet使用深度可分离卷积虽然减少了参数的数量,但是效果仍比不上同等规模的VGG等网络模型。
发明内容
针对上述不足,本发明提供了一种通用的基于嵌入式平台的行人检测模型的训练方法,其特征在于,包括:
(1)数据集的准备,进行图像标准化;
(2)训练浮点模型,采用降低浮点数位数进行深度学习模型训练;
(3)微调量化卷积层,使用上一步骤(2)得到的浮点模型,并将模型的路径添加到微调量化卷积脚本中进行微调,得到微调量化卷积层;
(4)应用步骤(3)线性变换微调的模型,利用标注的样本对预训练参数进行调整,以针对行人检测任务来修正模型网络;
(5)模型效果测试,利用测试集测试行人检测模型的准确率和召回率。
本发明创造的优点:针对嵌入式设备,利用特征提取与回归相分离的思路,并对SSD网络模型进行改进,实现了在嵌入式设备上对行人的实时检测。本发明提出的方法对采集图像的角度和环境亮度无严苛要求,使用灵活方便,检测的实时性好,精度高,可以很好地解决目前已有方法存在的问题。本方法与其它现有的在嵌入式设备上应用的行人检测训练方法相比,在实时性和检测的灵活性上有较大的提高,同时具有高准确率和召回率等良好性能。
附图说明
图1是训练流程示意图。
具体实施方式
下面结合附图对本发明作进一步描述,应当理解,此处所描述的内容仅用于说明和解释本发明,并不用于限定本发明。
行人检测在嵌入式设备上的应用是一项极具工程挑战的问题,嵌入式设备的优点在于使用方便体积小,结构简单,但其最大的缺点在于计算能力差,针对部分现有的神经网络模型难以达到实时的效果,且精度损失较大。以往的方法大多针对于某种特定的嵌入式平台进行设计,本发明旨在构建一种通用的基于嵌入式平台的行人检测模型的训练方法,通过如下步骤来解决:
(1)数据集的准备
本发明所使用的数据集为Caltech数据集,该数据集是由加州理工学院所提供的一个目前规模最大的行人数据集,其使用车载摄像头拍摄。图像的标准化是将图像中像素值减去各个通道的均值再除以各个通道的标准差,输入图像的标准化减少了网络输入数据的分布,使得网络更容易训练。在训练之前仍需对图像进行预处理,将图像的宽高分别更改尺寸(resize)到448像素。
(2)训练浮点模型
深度学习模型的计算任务分为训练和推理,训练过程往往是利用GPU强大的浮点计算能力来完成网络模型参数的学习过程。相对于训练过程而言,推理过程往往被应用于嵌入式终端设备,计算资源和功耗都受到了严格的限制。为了解决这样的问题,提出了很多不同的方法来减少模型的大小以及所需的计算资源。模型压缩除了剪枝以外,还有一个方法就是降低模型参数的精度。大多数的深度学习模型使用的是32位单精度浮点数来进行训练,本申请方法则通过降低到16位浮点数进行深度学习模型训练,从而减小了训练深度学习模型所需要的内存,在不损失大量精度的情况下加快了模型推理速度,从而也进一步提高了硬件效率。
本申请方法在caffe框架下进行训练,使用GPU进行模型训练。训练所使用的solver.prototxt,配置文件设置的参数内容如下所示。
在这一步骤,最终会得到snapshot_iter_150.caffemodel和snapshot_iter_150.solverstate两个文件。
(3)微调量化卷积层
通过实验结果表明,将浮点模型直接量化为12位的模型所造成的精度损失是可以忽略不计的。在该步骤中使用上一步骤得到的浮点模型,推理计算出对应的量化模型,量化模型中的卷积层为微调量化卷积层。
对于目标检测问题而言,通过持续的反向传播来微调经预训练后的模型权重,或者固定前面一些层以防止模型过拟合。通过实验可以观察到,模型的一些早期特征中所包含的特征更一般化,可以在许多任务中发挥作用,而模型更靠后的层会越来越特定于该目标任务所包含的分类的细节。
在将模型迁移到嵌入式设备时,由于参数共享的关系,可以在不同空间尺寸的图像上运行一个网络。这类似与于传统图像处理中的图像金字塔。图像金字塔是图像多尺度表达的一种方式,是一种以多分辨率来解释图像特征的典型结构。使用多特征尺度学习的方法可以提高模型的泛化效果。
本申请在开始时预先在Caltech数据集上训练得到了行人检测的模型。在后续的任务和处理中,均以此行人检测的模型为基础进行训练。数据集经过GPU训练计算后得到的模型只是针对GPU版本的浮点模型,这也是步骤(2)中所说明的浮点模型。当其应用在神经网络加速芯片时,需对其进行量化处理,其所用到的方法就是微调量化卷积层。微调量化卷积层需要将原来的浮点运算进行量化处理,降低数据所占用的存储空间,以较低的计算量去计算和预测所需要完成的任务。这也就是微调量化卷积层的内容,具体而言,通过截断预先训练好的网络的卷积层,并用经微调量化处理后的微调量化卷积层替换原有的卷积层。
普通预训练模型的特点是使用了大型数据集做训练,已经具备了提取浅层基础特征和深层抽象特征的能力。如果不做微调,模型需要重新训练,需要大量的数据、计算时间和资源,同时存在模型不收敛,参数不够优化,准确率低等风险。微调卷积层需要使用较小的学习率,通常使用比初始学习率小10倍的学习率。
(4)微调量化激活层
在卷积层对应微调量化后,对应的非线性激活层也需要微调量化。需要注意的是,这一步中将浮点模型直接在激活层上量化到12位时,也会相应的造成精度的损失。但是在实际实验验证的过程中发现,这些精度损失对最终的结果是可以忽略不计的。
对于目标检测问题而言,通过持续的反向传播来修改激活层中的非线性参数,以提高神经网络对数据的拟合效果。通过实验也可以观察到,神经网络能够起到强大的作用主要在于其非线性的学习能力。相应的,量化激活层是非常必要的。
(5)应用线性变换微调的模型
神经网络能发挥强大的学习能力主要是因为其包含非线性激活函数决定的。经过非线性激活层学习后,数据在对非线性问题上具有较好的拟合效果。但仅有此是不够的,每个神经网络包含若干个非线性单元。在经过非线性激活层激活后,需要应用全连接线性变换将各个微调后的量化激活层联系起来。
在做实际应用时,微调模型旨在利用已经标注的样本对参数进行调整,以针对当前的行人检测任务来修正网络。已标注的样本是步骤(1)中所说明的Caltech数据集中的图像数据。在该步骤中学习速率可以设置为较小的值,如0.00001。然后可以将得到的模型直接应用到嵌入式设备上。
(6)模型效果测试
召回率的计算公式为:
其中,P表示准确率,TP表示行人检测模型将包含行人的窗口预测为真的数量,FP表示行人检测模型将不包含行人的窗口预测为真的数量,FN表示行人检测模型将包含行人的窗口预测为假的数量。本申请的平均准确率为87%,平均召回率为91%。在所使用的特定嵌入式设备上同其他模型在速度和精度的比较如表1所示。
表1 在该应用场景下不同模型检测的速度和精度
MobileNet+NCNN | VGG16+NCNN | 本申请 | |
速度(FPS) | 3.3 | 2.9 | 10.1 |
精度(mAP) | 0.565 | 0.542 | 0.681 |
最后应说明的是:以上所述仅为本发明的解释,并不用于限制本发明,尽管对本发明进行了详细的说明,对于本领域的技术人员来说,其依然可以对前述所记载的技术方案进行修改,或者对其中部分技术特征进行等同替换。凡在本发明的精神和原则之内,所作的任何修改、等同替换、改进等,均应包含在本发明的保护范围之内。
Claims (9)
1.一种通用的基于嵌入式平台的行人检测模型的训练方法,其特征在于,包括:
(1)数据集的准备,进行图像标准化;
(2)训练浮点模型,采用降低浮点数位数进行深度学习模型训练;
(3)微调量化卷积层,使用上一步骤(2)得到的浮点模型,并将模型的路径添加到微调量化卷积脚本中进行微调,得到微调量化卷积层;
(4)应用步骤(3)线性变换微调的模型,利用已标注的样本对预训练参数进行调整,以针对行人检测任务来修正模型网络;
(5)模型效果测试,利用测试集测试行人检测模型的准确率和召回率。
2.根据权利要求1所述一种通用的基于嵌入式平台的行人检测模型的训练方法,其特征在于,步骤(1)中图像标准化是将图像中像素值减去各个通道的均值再除以各个通道的标准差。
3.根据权利要求1所述一种通用的基于嵌入式平台的行人检测模型的训练方法,其特征在于,步骤(1)中数据集为Caltech数据集,对图像进行预处理,将图像的宽高分别更改尺寸到448像素。
4.根据权利要求1所述一种通用的基于嵌入式平台的行人检测模型的训练方法,其特征在于,步骤(2)中通过16位浮点数进行深度学习模型训练。
5.根据权利要求1所述一种通用的基于嵌入式平台的行人检测模型的训练方法,其特征在于,步骤(2)中在caffe框架下进行训练,使用GPU进行模型训练。
6.根据权利要求1所述一种通用的基于嵌入式平台的行人检测模型的训练方法,其特征在于,步骤(3)中利用微调量化卷积层替换原有的卷积层。
7.根据权利要求1所述一种通用的基于嵌入式平台的行人检测模型的训练方法,其特征在于,步骤(4)中学习速率设置为0.00001,然后将得到的模型直接应用到嵌入式平台上。
9.根据权利要求1所述一种通用的基于嵌入式平台的行人检测模型的训练方法,其特征在于,步骤(3)中,对卷积层对应微调量化后,与之对应的非线性激活层也进行对应量化。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110883375.7A CN113343949B (zh) | 2021-08-03 | 2021-08-03 | 一种通用的嵌入式平台的行人检测模型训练方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110883375.7A CN113343949B (zh) | 2021-08-03 | 2021-08-03 | 一种通用的嵌入式平台的行人检测模型训练方法 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN113343949A true CN113343949A (zh) | 2021-09-03 |
CN113343949B CN113343949B (zh) | 2021-10-29 |
Family
ID=77480624
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202110883375.7A Active CN113343949B (zh) | 2021-08-03 | 2021-08-03 | 一种通用的嵌入式平台的行人检测模型训练方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN113343949B (zh) |
Citations (10)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN107273876A (zh) * | 2017-07-18 | 2017-10-20 | 山东大学 | 一种基于深度学习的‘宏to微转换模型’的微表情自动识别方法 |
CN107451658A (zh) * | 2017-07-24 | 2017-12-08 | 杭州菲数科技有限公司 | 浮点运算定点化方法及系统 |
CN109902745A (zh) * | 2019-03-01 | 2019-06-18 | 成都康乔电子有限责任公司 | 一种基于cnn的低精度训练与8位整型量化推理方法 |
CN109977782A (zh) * | 2019-02-27 | 2019-07-05 | 浙江工业大学 | 基于目标位置信息推理的跨店经营行为检测方法 |
CN110321874A (zh) * | 2019-07-12 | 2019-10-11 | 南京航空航天大学 | 一种轻量化的卷积神经网络行人识别方法 |
CN110472529A (zh) * | 2019-07-29 | 2019-11-19 | 深圳大学 | 目标识别导航方法及系统 |
CN110781912A (zh) * | 2019-09-10 | 2020-02-11 | 东南大学 | 一种基于通道扩张倒置卷积神经网络的图像分类方法 |
US20200082165A1 (en) * | 2016-12-16 | 2020-03-12 | Peking University Shenzhen Graduate School | Collaborative deep network model method for pedestrian detection |
CN112287839A (zh) * | 2020-10-29 | 2021-01-29 | 广西科技大学 | 一种基于迁移学习的ssd红外图像行人检测方法 |
CN113095470A (zh) * | 2020-01-08 | 2021-07-09 | 字节跳动有限公司 | 神经网络的训练方法、图像处理方法及装置、存储介质 |
-
2021
- 2021-08-03 CN CN202110883375.7A patent/CN113343949B/zh active Active
Patent Citations (10)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US20200082165A1 (en) * | 2016-12-16 | 2020-03-12 | Peking University Shenzhen Graduate School | Collaborative deep network model method for pedestrian detection |
CN107273876A (zh) * | 2017-07-18 | 2017-10-20 | 山东大学 | 一种基于深度学习的‘宏to微转换模型’的微表情自动识别方法 |
CN107451658A (zh) * | 2017-07-24 | 2017-12-08 | 杭州菲数科技有限公司 | 浮点运算定点化方法及系统 |
CN109977782A (zh) * | 2019-02-27 | 2019-07-05 | 浙江工业大学 | 基于目标位置信息推理的跨店经营行为检测方法 |
CN109902745A (zh) * | 2019-03-01 | 2019-06-18 | 成都康乔电子有限责任公司 | 一种基于cnn的低精度训练与8位整型量化推理方法 |
CN110321874A (zh) * | 2019-07-12 | 2019-10-11 | 南京航空航天大学 | 一种轻量化的卷积神经网络行人识别方法 |
CN110472529A (zh) * | 2019-07-29 | 2019-11-19 | 深圳大学 | 目标识别导航方法及系统 |
CN110781912A (zh) * | 2019-09-10 | 2020-02-11 | 东南大学 | 一种基于通道扩张倒置卷积神经网络的图像分类方法 |
CN113095470A (zh) * | 2020-01-08 | 2021-07-09 | 字节跳动有限公司 | 神经网络的训练方法、图像处理方法及装置、存储介质 |
CN112287839A (zh) * | 2020-10-29 | 2021-01-29 | 广西科技大学 | 一种基于迁移学习的ssd红外图像行人检测方法 |
Non-Patent Citations (1)
Title |
---|
刘键: "基于卷积神经网络的行人检测方法研究", 《中国优秀硕士学位论文全文数据库 信息科技辑(月刊)》 * |
Also Published As
Publication number | Publication date |
---|---|
CN113343949B (zh) | 2021-10-29 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN114120019B (zh) | 一种轻量化的目标检测方法 | |
CN108898168B (zh) | 用于目标检测的卷积神经网络模型的压缩方法和系统 | |
CN109241982B (zh) | 基于深浅层卷积神经网络的目标检测方法 | |
CN113052210B (zh) | 一种基于卷积神经网络的快速低光照目标检测方法 | |
CN113159073B (zh) | 知识蒸馏方法及装置、存储介质、终端 | |
CN108230354B (zh) | 目标跟踪、网络训练方法、装置、电子设备和存储介质 | |
CN110956126A (zh) | 一种联合超分辨率重建的小目标检测方法 | |
CN112348747A (zh) | 图像增强方法、装置及存储介质 | |
CN113191489B (zh) | 二值神经网络模型的训练方法、图像处理方法和装置 | |
CN113052185A (zh) | 一种基于Faster R-CNN的小样本目标检测方法 | |
Li et al. | A lightweight multi-scale aggregated model for detecting aerial images captured by UAVs | |
Zhang et al. | Dual-channel multi-task CNN for no-reference screen content image quality assessment | |
CN111178438A (zh) | 一种基于ResNet101的天气类型识别方法 | |
CN113569881A (zh) | 一种基于链式残差与注意力机制的自适应语义分割方法 | |
CN113298032A (zh) | 基于深度学习的无人机视角图像的车辆目标检测方法 | |
CN111950700A (zh) | 一种神经网络的优化方法及相关设备 | |
CN114332942A (zh) | 基于改进YOLOv3的夜间红外行人检测方法及系统 | |
CN112215334A (zh) | 一种面向事件相机的神经网络模型压缩方法 | |
CN111915558A (zh) | 一种高压输电线销钉状态检测方法 | |
CN114972753A (zh) | 基于上下文信息聚合和辅助学习的轻量级语义分割方法及系统 | |
CN104537624A (zh) | 基于ssim校正聚类稀疏表示的sar图像降斑方法 | |
CN113963333A (zh) | 一种基于改进yolof模型的交通标志牌检测方法 | |
CN113343949B (zh) | 一种通用的嵌入式平台的行人检测模型训练方法 | |
CN112825116B (zh) | 监控视频图像人脸检测和追踪方法、装置、介质及设备 | |
CN116129417A (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 | ||
GR01 | Patent grant | ||
GR01 | Patent grant |