Processing math: 100%
  • 中国精品科技期刊
  • CCF推荐A类中文期刊
  • 计算领域高质量科技期刊T1类
高级检索

基于上下文增强和特征提纯的小目标检测网络

肖进胜, 赵陶, 周剑, 乐秋平, 杨力衡

肖进胜, 赵陶, 周剑, 乐秋平, 杨力衡. 基于上下文增强和特征提纯的小目标检测网络[J]. 计算机研究与发展, 2023, 60(2): 465-474. DOI: 10.7544/issn1000-1239.202110956
引用本文: 肖进胜, 赵陶, 周剑, 乐秋平, 杨力衡. 基于上下文增强和特征提纯的小目标检测网络[J]. 计算机研究与发展, 2023, 60(2): 465-474. DOI: 10.7544/issn1000-1239.202110956
Xiao Jinsheng, Zhao Tao, Zhou Jian, Le Qiuping, Yang Liheng. Small Target Detection Network Based on Context Augmentation and Feature Refinement[J]. Journal of Computer Research and Development, 2023, 60(2): 465-474. DOI: 10.7544/issn1000-1239.202110956
Citation: Xiao Jinsheng, Zhao Tao, Zhou Jian, Le Qiuping, Yang Liheng. Small Target Detection Network Based on Context Augmentation and Feature Refinement[J]. Journal of Computer Research and Development, 2023, 60(2): 465-474. DOI: 10.7544/issn1000-1239.202110956
肖进胜, 赵陶, 周剑, 乐秋平, 杨力衡. 基于上下文增强和特征提纯的小目标检测网络[J]. 计算机研究与发展, 2023, 60(2): 465-474. CSTR: 32373.14.issn1000-1239.202110956
引用本文: 肖进胜, 赵陶, 周剑, 乐秋平, 杨力衡. 基于上下文增强和特征提纯的小目标检测网络[J]. 计算机研究与发展, 2023, 60(2): 465-474. CSTR: 32373.14.issn1000-1239.202110956
Xiao Jinsheng, Zhao Tao, Zhou Jian, Le Qiuping, Yang Liheng. Small Target Detection Network Based on Context Augmentation and Feature Refinement[J]. Journal of Computer Research and Development, 2023, 60(2): 465-474. CSTR: 32373.14.issn1000-1239.202110956
Citation: Xiao Jinsheng, Zhao Tao, Zhou Jian, Le Qiuping, Yang Liheng. Small Target Detection Network Based on Context Augmentation and Feature Refinement[J]. Journal of Computer Research and Development, 2023, 60(2): 465-474. CSTR: 32373.14.issn1000-1239.202110956

基于上下文增强和特征提纯的小目标检测网络

基金项目: 国家自然科学基金青年科学基金项目(42101448 );中国科学院光电信息处理重点实验室开放课题基金项目(OEIP-O-202009)
详细信息
    作者简介:

    肖进胜: 1975年生.博士,副教授.CCF会员.主要研究方向为图像处理和计算机视觉

    赵陶: 1996年生.硕士.主要研究方向为图像处理和计算机视觉

    周剑: 1989年生.博士,博士后.主要研究方向为移动视觉测量和高精度地图

    乐秋平: 1997年生.硕士.主要研究方向为图像处理和计算机视觉

    杨力衡: 1995年生.硕士研究生.主要研究方向为图像处理和计算机视觉

    通讯作者:

    周剑(jianzhou @whu.edu.cn

  • 中图分类号: TP183

Small Target Detection Network Based on Context Augmentation and Feature Refinement

Funds: This work was supported by the National Natural Science Foundation of China for Young Scientists (42101448) and the Open Project Program Foundation of the CAS Key Laboratory of Opto-Electronics Information Processing (OEIP-O-202009).
  • 摘要:

    微小目标的纹理模糊、包含特征少,是目标检测领域的难点.针对小目标检测提出一种新的上下文增强模块(context augmentation module, CAM)和特征提纯模块(feature refinement module, FRM)相结合的特征金字塔复合结构. 利用多尺度空洞卷积的特征融合,补充网络中的上下文信息;引入通道和空间的特征提纯机制来抑制多尺度特征融合后的冲突信息,防止小目标淹没在冲突信息中;同时,引入复制—缩小—粘贴(copy-reduce-paste)的数据增强方法提高小目标的占比,使训练时小目标对损失值的贡献更大,训练更加平衡.由实验结果可知,所提出的算法在VOC数据集上目标检测的平均精度均值(Mean Average Precision, mAP)达到了83.6%(交并比为0.5);对小目标检测的AP值达到了16.9%(交并比为0.5~0.95),比YOLOV4,CenterNet,RefineDet的分别提高3.9%,7.7%和5.3%.在TinyPerson数据集上小目标检测的AP值为55.1%,比YOLOV5,DSFD的分别提高0.8%和 3.5%.

    Abstract:

    Small objects contain few and fuzzy features, which is a hard problem in the field of object detection. The poor performance of small object detection is mainly caused by the limitation of the network and the imbalance of the training dataset. A novel feature pyramid composite structure constructed by context augmentation module (CAM) and feature refinement module (FRM) is proposed. The feature fusion of multi-scale dilated convolution is applied to generate features on different receptive fields, and then the features are added to detection network to supplement context information. The channel and space feature refinement mechanism is introduced to suppress the conflict information generated by multi-scale feature fusion and prevent small objects from being submerged in the conflict information. Besides, a data augmentation method called copy-reduce-paste is proposed to increase the proportion of small targets, so that the contribution of small targets to the loss value during training is greater and the training is more balanced. Experimental results show that the Mean Average Precision(mAP) of object detection on the VOC dataset of the proposed network is 83.6% (IOU is 0.5). The AP value of small target detection is 16.9% (IOU changes from 0.5 to 0.95), which is 3.9%, 7.7% and 5.3% higher than that of YOLOV4, CenterNet and RefineDet, respectively. The AP value of small target detection on TinyPerson dataset is 55.1%, which is 0.8% and 3.5% higher than that of YOLOV5 and DSFD, respectively.

  • 国密SM4算法[1]是一种常用的分组密码算法,广泛应用于数据保护、加密通信等领域. SM4算法常见工作模式有ECB(electronic codebook),CBC(cipher block chaining)等,对于相同的明文块,ECB模式下会产生完全相同的密文,而在CBC模式下,当前的明文块会与前一块的密文异或后进行运算. 因此,即使是完全相同的明文输入也可能会有完全不同的密文输出. 相比于ECB模式,CBC模式提供了更高的安全性和抵抗攻击的能力,有着更高的应用需求. 提高SM4算法在CBC模式下的性能,对于在边缘设备中使用SM4算法是至关重要的. 但是,在CBC模式下存在着难以提高吞吐率的问题:每组的输入必须等待前一组运算结束后才能获得,因而难以使用流水线方法提升吞吐率.

    文献[2]中提到了一种改进方法,将电路中的S盒以外的其他逻辑结构进行预计算,并把预计算的结果与S盒进行融合构成新的查找表,从而提高SM4算法在CBC模式下吞吐率. 本文基于此工作进一步优化了S盒的表示,并针对轮函数的迭代过程进行了优化,最终减少了轮函数关键路径上的2次异或运算,有效提高了算法的性能.

    本文的设计针对CBC模式下的SM4算法,在TSMC 40 nm,SMIC 55 nm工艺下,使用Synopsys Design Compiler分别进行了ASIC综合. 综合结果显示,本文所提出的设计在CBC模式下的吞吐率达到了4.2 Gb/s,同时单位面积吞吐量达到了129.4 Gb·s−1·mm−2,明显优于已发表的类似设计. 这些结果表明本文所提出的化简方法在改进SM4算法性能方面具有很大的潜力.

    本文的结构为:首先介绍了SM4算法及其在CBC模式下存在的性能瓶颈问题. 然后,详细描述了本文提出的2个化简方法,并解释了它们在轮函数迭代和S盒置换过程中的作用. 接下来,介绍了实验设计并给出了实验结果分析和对比. 最后,对进一步改进和应用的方向进行了展望.

    SM4算法是一种对称密钥密码算法,被广泛应用于数据加密和保护领域,它是中国密码算法的标准之一,具有较高的安全性和良好的性能.

    SM4采用了分组密码的设计思想,将明文数据划分为128 b的数据块,并通过密钥对每个数据块进行加密和解密操作. 对单组数据进行加解密的流程如图1所示,分为密钥扩展算法和加解密算法2部分. 图1中的FK是系统预设的参数,与用户密钥进行异或运算后作为密钥扩展算法的输入. 加解密算法接受密钥扩展算法产生的32轮轮密钥rki对明文进行加解密,最后经反序变换输出. 加解密使用的是同一套计算流程,唯一的区别是解密时使用轮密钥的顺序与加密过程相反.

    图  1  SM4算法工作流程
    Figure  1.  Workflow of SM4 algorithm

    密钥扩展算法和加解密算法2部分均由32次轮函数迭代构成,整体结构均采用4路并行的Feistel结构,在计算过程中,以128 b数据为输入、128 b数据为输出,其内部的运算逻辑如图2所示. 输出中的前96 b数据等于输入中的后96 b数据,输出后的32 b数据通过轮函数运算产生.

    图  2  4路并行的Feistel结构
    Figure  2.  Four parallel Feistel structure

    在密钥扩展算法中使用的密钥是算法给定的固定密钥,记作cki. 在加解密算法中使用的密钥是由密钥扩展算法通过用户给的密钥扩展出来的轮密钥,记作rki.

    SM4密钥扩展算法结构如图3所示,密钥扩展的主要过程包括32轮密钥扩展的轮函数,其中,密钥为128 b,FK为SM4标准中规定的一个128 b常数. 二者异或后的值将会作为密钥扩展轮函数的首轮输入,并通过一个选择器进行循环迭代,总计迭代32轮产生32个轮密钥.

    图  3  SM4的密钥扩展算法结构
    Figure  3.  Key expansion algorithm structure of SM4

    设用户输入的密钥为MK,则该密钥对应的32轮轮密钥可以按照式(1)求出:

    {(k0,k1,k2,k3)=MKFK,ki+4=kiF(ki+1ki+2ki+3cki),rki=ki+4, (1)

    其中,cki是系统预设的32 b参数,rki代表第i轮的轮密钥,F代表密钥扩展轮函数,其由S盒置换算法τ:Z322Z322和线性变换算法L(x)=x(x<<<13)(x<<<23)组成,其中<<<表示循环左移运算.

    SM4算法的加解密算法的整体结构与密钥扩展算法类似,均包含32轮的轮函数迭代,区别在于加解密算法中额外包含1次反序变换.

    SM4算法的轮函数迭代流程如图4所示,X1~X4为第1轮的输入,X2~X5为第1轮的输出,同时也是第2轮的输入. rk1为第1轮的轮密钥,T函数代表加解密模块的轮函数. 与密钥扩展部分的轮函数F类似,由S盒置换算法τ和一个线性变换算法L(x)=x(x<<<2)(x<<<10) (x<<<18)(x<<<24)组成.

    图  4  SM4加解密模块轮函数结构
    Figure  4.  Round function structure of SM4 encryption and decryption modules

    通过多轮的迭代过程,SM4算法能够实现高强度的数据加密和解密. 然而,在CBC模式下,由于相邻数据之间的依赖关系,传统的流水线技术难以提高算法的吞吐率. 因此,针对这一问题,本文提出了2种化简方法,以减少关键路径上的运算,从而提高SM4算法在CBC模式下的性能.

    加解密模块的轮函数的结构如图4所示,若不考虑T函数带来的时序延迟,单次轮函数迭代的关键路径上共包含3次异或运算. 以公式的形式描述SM4算法加解密轮函数的迭代关系可得到式(2):

    Xi+4=Xi(Xi+1Xi+2Xi+3rki). (2)

    若考虑相邻的2次轮函数迭代,则有:

    {Xi+4=XiT(Xi+1Xi+2Xi+3rki),Xi+5=XiT(Xi+2Xi+3Xi+4rki+1). (3)

    观察式(1)~(3)不难发现,由于SM4采用了4条数据线路的Feistel结构进行设计,在相邻的2次轮函数迭代过程中,均有96 b的输入是完全一致的,在式(3)的计算过程中,相邻2轮的轮函数将Xi+2Xi+3计算了2次.

    因此,一个简单的优化思路便是,我们在轮函数之间传递数据时,额外传递Xi+2Xi+3rki+1的运算结果,并作用于下一次计算,得到的流程图如图5所示.

    图  5  优化的轮函数结构
    Figure  5.  Optimized round function structure

    相比于图4的运算流程,在计算当前轮次的输出时,二次优化过后的轮函数通过提前获取下一轮次使用的密钥,并利用2轮之间相同的数据提前计算,可以使得在加解密的流程中总计节省32次异或运算的时间.

    S盒是密码学领域的一个基本组件,其功能是实现数据的非线性变换,在DES,AES,SM1,SM4等算法中均有应用. 在SM4算法中,其提供了一个8 b到8 b的非线性变换.

    在SM4算法中,S盒模块通常与另一个线性变换函数L组合使用,即图4图5中的T函数,其位于加解密算法轮函数的关键路径上,因此,如果能找到优化T函数关键路径的方法延时,也可以使得整个加解密模块的延时变小,进而提高运算效率.T函数的内部结构如图6所示,图中的<<<表示对32 b数据进行循环左移,关键路径包括1个S盒和3次异或运算. 在硬件实现中,循环移位可以通过硬件连线来实现,不会带来额外的路径延时.

    图  6  SM4加解密模块T函数结构
    Figure  6.  T function structure of SM4 encryption and decryption modules

    T函数中包含4次异或运算,反映到电路设计中,其关键路径上至少存在3次异或运算. 因此,一个优化思路便是,将算法中的S盒的输入输出修改为8 b输入、32 b输出[2-3] ,并提前将L函数作用于图中的4个S盒,如图7所示. 图7中,通过编码的形式保存其运行结果,将图6中的SBox与后续的线性变换L组合形成exSBox,之后仅需要将4个exSBox的输出异或即可,从而减少了1次异或运算.

    图  7  优化的T函数结构
    Figure  7.  Optimized T-function structure

    虽然修改后的S盒比原先的S盒输出了更多的数据,但在硬件实现中,仍然是通过相同数量的多路选择器查表输出. 因此修改前后的S盒的路径延时及其安全性并未改变.

    图7中的exSBox1为例,使用0xff作为输入展示exSBox1的构造方式,首先获得0xff作用于S盒后的运行结果0x48. 由于exSBox1的输入对应最高四位,因此,将其拓展为32 b数据为0x48000000. 在经过L函数后,得到的值是0x68492121. 如表1所示,表中前5行加粗部分表示传入的数据及其循环移位后所处位置,其余位置在任意输入下都恒等于0.

    表  1  搜索空间降低比率和命中率
    Table  1.  Search Space Reduction Rate and Hit Rate
    原数据 01001000 00000000 00000000 00000000
    <<<2 00100000 00000000 00000000 00000001
    <<<10 00000000 00000000 00000001 00100000
    <<<18 00000000 00000001 00100000 00000000
    <<<24 00000000 01001000 00000000 00000000
    异或和 01101000 01001001 00100001 00100001
    注:加粗部分表示传入的数据及其循环移位后所处位置.
    下载: 导出CSV 
    | 显示表格

    观察表1的运算结果不难发现,除最后一行加粗数字表示的第0~5位,第14,15位由异或运算产生,其余的24位均是输入的8位数据的排列组合,因此在硬件设计时,可以仅使用8 b输入、16 b输出的S盒实现. 对于图7中剩余的3个exSBox,在相同的输入下,可以通过对表1中的数据进行循环移位,得到对应的输出. 上述结论对4个位于不同部位的S盒均成立.

    具体而言,令p为输入的8 b数据,τ(p)为标准SM4算法中S盒的输出. X=(x0,x1,,x15)为exSBox1中存储的16 b数据,Y=(y0,y1,,y31)为优化后的T函数中需要的32 b输出. τ为SM4算法标准中使用的S盒置换函数,其对于8 b输入,产生对应的8 b输出,则X可以由式(4)产生:

    {(x0,x1,,x7)=τ(p),(x8,x9,,x15)=τ(p)(τ(p)<<<2). (4)

    表1可知,Y的取值实际上可以由X经过排列组合得到,对于exSBox2,exSBox3,exSBox4的取值,可以通过Y循环移位得到,且由于该过程中仅包含赋值运算,在电路设计中可以通过物理连线完成. 相比于文献[2]中的设计,节约了1/3的面积消耗. 具体的计算方式如式(5)所示.

    {(y0,y1,,y5)=(x8,x9,,x13)(y6,y7=(x6,x7)(y8,y9,,y13)=(x0,x1,,x5)(y14,y15=(x14,x15)(y16,y17,,y21)=(x2,x3,,x7)(y22,y23=(x0,x1)(y24,y25,,y29)=(x2,x3,,x7)(y30,y31=(x0,x1). (5)

    现场可编程逻辑门阵列(FPGA)和专用集成电路(ASIC)是目前主流使用硬件电路实现密码算法的2个方式. FPGA虽然具有可编程性、灵活性和快速设计等优势,但ASIC相较于FPGA拥有更高的性能,与本文设计追求的高效率目标相符,所以选择在ASIC下实现.

    SM4硬件系统的整体结构设计如图8所示,包括密钥扩展模块、加解密模块和适配CBC工作模式的组合逻辑. 对于单个加解密任务,若明文被分为n组,会执行1次密钥扩展和n次加解密. 因此,优化加解密算法的执行效率是优化SM4硬件设计的重点. 本文所提出的2种化简方法,对于每一组明文输入,可以减少64级异或门的延时,极大地提升了运算效率.

    图  8  SM4硬件整体架构
    Figure  8.  Overall architecture of SM4 hardware

    SM4算法的硬件实现主要有2种方案:一种方案是流水线结构,即通过寄存器连接多个加解密模块同时工作以提高加解密的效率,如图9(a)所示;另一种方案是使用循环迭代的方式. 即一次性提取32个轮函数中的n轮组合成一个组合电路,称为n合1电路,如图9(b)所示. 流水线结构的优势是可以充分利用n个加密核心的性能,在不影响整体工作频率的情况下加速运算. 对于SM4算法而言,在合理范围内堆叠流水线可以实现极高的吞吐量.

    图  9  流水线结构与循环迭代结构
    Figure  9.  Pipeline architecture and loop iteration architecture

    然而,流水线结构仅适用于ECB等数据无前后依赖的工作模式. 在CBC工作模式下,由于需要将前一轮的输出与本轮的输入进行异或运算,相邻的数据存在依赖,故而无法使用流水线加速运算. 因此,在本设计中没有选用流水线结构.

    虽然循环迭代结构会降低整体模块的工作频率,对吞吐量的提升较为有限,但可以同时兼容 ECB,CBC这 2种工作模式. 本设计最终选择了循环迭代的设计方式.

    在SM4算法中,密钥扩展与加解密算法类似,均包含32轮迭代. 密钥扩展模块采用图2所示的单轮组合逻辑电路循环32次来实现32轮迭代.

    在密钥扩展模块的输出端,使用寄存器存放每一轮电路的轮密钥,标号为0~31,如图10所示. 标号从0开始的好处是:在解密时,使用到的密钥顺序相反的,加密的第k轮使用的是第k1号密钥,解密的第k轮使用的是第32k号密钥. 在二进制下,二者的标号可以通过取反操作相互转化.

    图  10  轮密钥的存储与使用
    Figure  10.  Storage and usage of round keys

    为了保证运算结果的准确性,密钥扩展模块还 会向加解密模块发出控制信号表明自己的工作状态,以避免在轮密钥尚未完全更新时使用错误的轮密钥进行加解密.

    在国家标准文档[1]中,并没有针对CBC工作模式给出具体的测试用例. 因此,本文设计方案通过完整的Verilog HDL语言实现,通过在FPGA平台进行综合、仿真和上板验证,以确保功能正确并进行相关性能分析,如图11所示. 具体而言,通过PCIE上位机下发随机的明文数据到FPGA开发板,开发板完成加密后传回上位机,通过与软件对比实现功能验证. 若在循环验证多次后二者的输出均完全相同,则认为设计的SM4电路的功能正确.

    图  11  测试流程
    Figure  11.  Testing procedures

    最终,本文的设计在Zynq 7020 FPGA开发板上完成了上板验证,确保了功能的正确性,工作频率最高可达95 MHz,吞吐量约为1.5 Gb/s.

    ASIC上主要针对2种工艺SMIC 55 nm与 TSMC 40 nm进行了测试、通过Synopsys公司的EDA工具DesignCompiler进行时序等综合约束,我们选择了芯片面积和芯片使用的逻辑门数量(gates)作为评估指标,其结果如表2表3所示,在CBC模式下,本文的设计在3.97 mW的功耗下,单位面积吞吐率达129.4 Gb·s−1·mm−2,明显优于同类设计. 此外,以使用逻辑门的数量为评估标准,本文提出的设计在该指标上也明显优于同类设计,单位面积吞吐率为0.205×10−3 Gb·s−1·gates−1.

    表  2  SM4综合结果与面积效率对比
    Table  2.  Comparison of SM4 Synthesis Results and Area Efficiency
    工艺节点 芯片面积/mm2 吞吐率/(Gb·s−1 单位面积吞吐率/
    (Gb·s−1·mm−2
    功耗/mW
    40 nm* 0.0335 4.34 129.40 3.97
    55 nm* 0.0877 4.41 50.30 10.88
    65 nm[2] 0.1260 5.24 41.59
    180 nm[4] 0.0790 0.10 1.27 5.31
    55 nm[5] 0.0870 0.40 4.59 4.35
    350 nm[6] 0.0270 0.412 15.26
    注:*标注的表示本文的结果.
    下载: 导出CSV 
    | 显示表格
    表  3  SM4综合结果与门效率对比
    Table  3.  Comparison of SM4 Synthesis Results and Gates Efficiency
    工艺节点 gates 吞吐率/(Gb·s−1 单位面积吞吐率/
    (Gb·s−1·gates−1
    40 nm* 21.2×103 4.34 0.205×10−3
    55 nm* 21.1×103 4.41 0.209×10−3
    180 nm[6] 32.0×103 0.80 0.025×10−3
    65 nm[7] 31.0×103 1.23 0.040×10−3
    55 nm[8] 22.0×103 0.27 0.012×10−3
    130 nm[9] 22.0×103 0.80 0.036×10−3
    注:*标注的表示本文的结果.
    下载: 导出CSV 
    | 显示表格

    在不同工艺、电压下对该设计进行综合,可以得到本文设计在不同使用场景下的吞吐率. 在TSMC 40 nm、SMIC 55 nm、SMIC 130 nm下使用不同的工艺角分别对本文的设计进行综合,结果如表4所示.

    表  4  不同工艺角下的SM4综合结果与效率对比
    Table  4.  Comparison of SM4 Synthesis Results and Efficiency with Different Process Corners
    工艺节点 工艺角 面积/gates 吞吐率/(Gb·s−1 功耗/mW
    40 nm 0.99V/125°C/SS 21.0×103 2.40 2.55
    1.1V/25°C/TT 21.2×103 4.34 3.97
    1.21V/0°C/FF 20.9×103 6.96 8.35
    55 nm 1V/25°C/TT 20.0×103 2.78 4.10
    1.2V/25°C/TT 21.1×103 4.41 10.88
    1.32V/0°C/FF 17.8×103 6.84 33.59
    130 nm 1.08V/125°C/SS 20.8×103 1.11 6.86
    1.2V/25°C/TT 21.0×103 1.75 15.70
    1.32V/0°C/FF 21.8×103 2.45 23.03
    下载: 导出CSV 
    | 显示表格

    根据本文提出的2种对SM4加解密模块关键路径进行化简以及降低面积的方法,实现了4合1的SM4电路,并基于Zynq7020开发板进行了功能验证. 此外,ASIC综合结果表明本文的SM4电路相比于其他方案有更高的单位面积吞吐率和更低的功耗. 因此,这种对SM4算法进行的优化是有效的,并且对其他分组算法提高CBC模式下的单位面积吞吐率具有参考价值.

    作者贡献声明:郝泽钰提出研究方案并完成了论文的撰写;代天傲、黄亦成、段岑林协助完成了ASIC平台上的验证实验;董进、吴世勇、张博、王雪岩、贾小涛提出指导意见并修改论文;杨建磊提出指导意见并讨论定稿.

  • 图  1   FPN总体网络结构

    Figure  1.   Overall network structure of FPN

    图  2   CAM结构图

    Figure  2.   The structure of CAM

    图  3   上下文信息增强效果图

    Figure  3.   Context information augmentation effect diagrams

    图  4   FRM结构

    Figure  4.   The structure of FRM

    图  5   FRM可视化结果

    Figure  5.   Visualization results of FRM

    图  6   数据增强示例

    Figure  6.   Data augmentation examples

    图  7   损失曲线

    Figure  7.   The curve of loss

    图  8   训练特征图可视化效果

    Figure  8.   Visualization results of feature maps in training

    表  1   CAM消融实验结果

    Table  1   Ablation Experimental Results of CAM %

    算法AP(IOU=0.5)AR(IOU=0.5)
    APsAPmARsARm
    基线模型34.860.557.978.7
    相加融合35.663.060.581.8
    自适应融合36.063.158.981.0
    拼接融合36.661.059.879.5
    注:基线模型为YOLOV3,测试数据集为VOC,IOU为交并比. APsAPm分别指小目标、中目标的平均精度;ARsARm分别指小目标、中目标的平均召回率.
    下载: 导出CSV

    表  2   VOC数据集目标尺寸统计结果

    Table  2   Statistical Results of Target Size on VOC Datebase %

    统计值小目标中目标大目标
    目标数量占比10.016.673.4
    图片数量占比8.216.275.6
    下载: 导出CSV

    表  3   数据增强消融实验结果

    Table  3   Ablation Experimental Results of Data Augmentation %

    粘贴次数AP(IOU=0.5)AR(IOU=0.5)
    APsAPmARsARm
    0(基线模型)34.860.557.978.7
    137.362.759.880.9
    236.862.658.081.0
    333.259.758.079.8
    注:基线模型为YOLOV3IOU为交并比.APsAPm分别指小目标、中目标的平均精度;ARsARm分别指小目标、中目标的平均召回率.
    下载: 导出CSV

    表  4   TinyPerson数据集上的检测结果

    Table  4   Detection Results on TinyPerson Dataset %

    算法主干网络APs
    MaskR-CNN[20]ResNet5042.5
    AL-MDN[21]AGG1634.1
    DSFD[22]ResNet15251.6
    YOLOV5[23]CSPDarkNet54.3
    本文算法Darknet5355.1
    注:APs指小目标的平均精度.
    下载: 导出CSV

    表  5   VOC数据集上的小目标检测结果

    Table  5   Results of Small Targets Detection on VOC Dataset %

    算法 AP(IOU[0.5,0.95]) AR(IOU[0.5,0.95])
    APsAPmARsARm
    RefineDet[10]11.634.920.239.9
    CenterNet[24]9.231.317.443.0
    YOLOV4[16]13.034.518.142.8
    本文算法16.933.429.445.8
    注:IOU为交并比.APsAPm分别指小目标、中目标的平均精度;ARsARm分别指小目标、中目标的平均召回率.
    下载: 导出CSV

    表  6   VOC数据集上的实验结果(IOU=0.5)

    Table  6   Experimental Results on VOC Dataset (IOU=0.5)

    类型算法主干网络输入尺寸mAP/%
    二阶段
    Faster R-CNN[4]ResNet1011000×60076.4
    R-FCN[3]ResNet1011000×60080.5
    HyperNet[25]VGG161000×60076.3
    CoupleNet[26]ResNet1011000×60082.7
    Reconfig[27]ResNet1011000×60082.4
    IPG-Net[28]IPGNet1011000×60084.8
    一阶段
    SSD[2]VGG16512×51279.8
    RefineDet[10]VGG16512×51281.8
    RFBNet[29]VGG16512×51282.2
    ScratchDet[30]RestNet34320×32080.4
    PFPNet[31]VGG16512×51282.3
    本文算法Darknet53448×44883.6
    本文算法+Darknet53448×44885.1
    注:“+”表示多尺度测试.
    下载: 导出CSV

    表  7   消融实验结果

    Table  7   Ablation Experimental Results

    基线模型增强CAMFRMAP/% (IOU = 0.5)AR/% (IOU = 0.5)
    APsAPmAPlARsARmARl
    34.860.583.657.978.792.8
    37.362.783.459.880.993.0
    36.661.084.259.879.593.1
    37.662.183.959.079.192.6
    40.264.184.664.881.093.9
    注:√表示包含该模块,IOU为交并比.APsAPmAPl分别指小目标、中目标和大目标的平均精度;ARsARmARl分别指小目标、中目标和大目标的平均召回率.
    下载: 导出CSV
  • [1]

    Joseph R, Ali F. YOLOV3: An incremental improvement[J]. arXiv preprint, arXiv: 1804.02767, 2018

    [2]

    Liu Wei, Anguelov D, Erhan D, et al. SSD: Single shot multi-box detector[C] //Proc of the 14th European Conf on Computer Vision. Berlin: Springer, 2016: 21−37

    [3]

    Girshick R, Donahue J, Darrell T, et al. Rich feature hierarchies for accurate object detection and semantic segmentation[C] //Proc of the 27th IEEE Conf on Computer Vision and Pattern Recognition. Piscataway, NJ: IEEE, 2014: 580−587

    [4]

    Ren Shaoqing, He Kaiming, Girshick R. Faster R-CNN: Towards real-time object detection with region proposal networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2017, 39(6): 1137−1149 doi: 10.1109/TPAMI.2016.2577031

    [5]

    Kisantal M, Wojna Z, Murawski J, et al. Augmentation for small object detection[J]. arXiv preprint, arXiv: 1902.07296, 2019

    [6] 黄继鹏,史颖欢,高阳. 面向小目标的多尺度Faster-RCNN检测算法[J]. 计算机研究与发展,2019,56(2):319−327 doi: 10.7544/issn1000-1239.2019.20170749

    Huang Jipeng, Shi Yinghuan, Gao Yang. Multi-scale faster-rcnn algorithm for small object detection[J]. Journal of Computer Research and Development, 2019, 56(2): 319−327 (in Chinese) doi: 10.7544/issn1000-1239.2019.20170749

    [7]

    Lin Tsungyi, Dollár P, Girshick R, et al. Feature pyramid networks for object detection[C] //Proc of the 30th IEEE Conf on Computer Vision and Pattern Recognition. Piscataway, NJ: IEEE, 2017: 936−944

    [8]

    Deng Chunfang, Wang Mengmeng, Liu Liang, et al. Extended feature pyramid network for small object detection[J]. arXiv preprint, arXiv: 2003.07021, 2020

    [9] 肖进胜,饶天宇,贾茜,等. 改进的自适应冲击滤波图像超分辨率插值算法[J]. 计算机学报,2015,38(6):1131−1139 doi: 10.11897/SP.J.1016.2015.01131

    Xiao Jinsheng, Rao Tianyu, Jia Qian, et al. Interpolation algorithm based on improved adaptive shock filter in image super-resolution[J]. Chinese Journal of Computers, 2015, 38(6): 1131−1139 (in Chinese) doi: 10.11897/SP.J.1016.2015.01131

    [10]

    Zhang Shifeng, Wen Longyin, Bian Xiao, et al. Single-shot refinement neural network for object detection[C] //Proc of the 31st IEEE/CVF Conf on Computer Vision and Pattern Recognition. Piscataway, NJ: IEEE, 2018: 4203−4212

    [11] 蒋弘毅,王永娟,康锦煜. 目标检测模型及其优化方法综述[J]. 自动化学报,2021,47(6):1232−1255 doi: 10.16383/j.aas.c190756

    Jiang Hongyim Wang Yongjuan, Kang Jingyu. A survey of object detection models and its optimization method[J]. Acta Automatica Sinica, 2021, 47(6): 1232−1255 (in Chinese) doi: 10.16383/j.aas.c190756

    [12]

    Liu Shu, Qi Lu, Qin Haifang, et al. Path aggregation network for instance segmentation[C] //Proc of the 31st IEEE/CVF Conf on Computer Vision and Pattern Recognition. Piscataway, NJ: IEEE, 2018: 8759−8768

    [13]

    Ghiasi G, Lin Tsungyi, Le Q V. NAS-FPN: Learning scalable feature pyramid architecture for object detection[C] //Proc of the 32nd IEEE/CVF Conf on Computer Vision and Pattern Recognition. Piscataway, NJ: IEEE, 2019: 7029−7038

    [14]

    Tan Mingxing, Pang Ruoming, Le Q V. EfficientDet: Scalable and efficient object detection[C] //Proc of the 33rd IEEE/CVF Conf on Computer Vision and Pattern Recognition. Piscataway, NJ: IEEE, 2020: 10778−10787

    [15]

    Chen Yukang, Zhang Peizhen, Li Zeming, et al. Stitcher: Feedback-driven data provider for object detection[J]. arXiv preprint, arXiv: 2004.12432, 2020

    [16]

    Bochkovskiy A, Wang Chienyao, Mark Liao H Y. YOLOV4: Optimal speed and accuracy of object detection[J]. arXiv preprint, arXiv: 2004.10934, 2020

    [17]

    Yu F, Vladlen K. Multi-scale context aggregation by dilated convolutions[J]. arXiv preprint, arXiv: 1511.07122, 2015

    [18] Yu Xuehui, Gong Yuqi, Jiang Nan, et al. Scale match for tiny person detection[C] //Proc of 2020 IEEE Winter Conf on Applications of Computer Vision. Piscataway, NJ: IEEE, 2020: 1257−1265
    [19]

    Paszke A, Gross S, Massa F, et al. Pytorch: An imperative style, high-performance deep learning library[C] //Proc of the 33rd Advances in Neural Information Processing Systems. Cambridge, MA: MIT Press, 2019: 8026−8037

    [20]

    He Kaiming, Gkioxari G, Dollárm P, et al. Mask R-CNN[C] //Proc of the 16th IEEE Int Conf on Computer Vision (ICCV). Piscataway, NJ: IEEE, 2017: 2980−2988

    [21]

    Choi J, Elezi I, Lee H J, et al. Active learning for deep object detection via probabilistic modeling[J]. arXiv preprint, arXiv: 2103.16130, 2021

    [22]

    Li Jian, Wang Yabiao, Wang Changan, et al. DSFD: Dual shot face detector[C] //Proc of the 32nd IEEE Conf on Computer Vision and Pattern Recognition. Piscataway, NJ: IEEE, 2019: 5060−5069

    [23]

    Jocher G. YOLOV5[CP/OL]. (2021-06-09) [2021-06-09]. https://github.com/ultralytics/yolov5

    [24]

    Duan Kaiwen, Bai Song, Xie Lingxi, et al. CenterNet: Keypoint triplets for object detection[C] //Proc of the 17th IEEE Int Conf on Computer Vision (ICCV). Piscataway, NJ: IEEE, 2019: 6568−6577

    [25]

    Kong Tao, Yao Anbang, Chen Yurong, et al. HyperNet: Towards accurate region proposal generation and joint object detection[C] //Proc of the 29th IEEE Conf on Computer Vision and Pattern Recognition (CVPR). Piscataway, NJ: IEEE, 2016: 845−853

    [26]

    Zhu Yousong, Zhao Chaoyang, Wang Jinqiao, et al. CoupleNet: Coupling global structure with local parts for object detection[C] //Proc of the 15th IEEE Int Conf on Computer Vision (ICCV). Piscataway, NJ: IEEE, 2017: 4146−4154

    [27]

    Kong Tao, Sun Fuchun, Huang Wenbing, et al. Deep feature pyramid reconfiguration for object detection[C] //Proc of the 15th European Conf on Computer Vision (ECCV). Berlin: Springer, 2018: 169−185

    [28]

    Liu Ziming, Gao Guangyu, Sun Lin, et al. IPG-net: Image pyramid guidance network for small object detection[C] //Proc of the 33rd IEEE/CVF Conf on Computer Vision and Pattern Recognition Workshops (CVPRW). Piscataway, NJ: IEEE, 2020: 4422−4430

    [29]

    Liu Songtao, Huang Di, Wang Yunhong. Receptive field block net for accurate and fast object detection[C] //Proc of the 15th European Conf on Computer Vision. Berlin: Springer, 2018: 404−419

    [30]

    Zhu Rui, Zhang Shifeng, Wang Xiaobo. ScratchDet: Training single-shot object detectors from scratch[C] //Proc of the 32nd IEEE/CVF Conf on Computer Vision and Pattern Recognition (CVPR). Piscataway, NJ: IEEE, 2019: 2263−2272

    [31]

    Kim S W, Kook H K, Sun J Y, et al. Parallel feature pyramid network for object detection[C] //Proc of the 15th European Conf on Computer Vision (ECCV). Berlin: Springer, 2018: 234−250

  • 期刊类型引用(1)

    1. 石明丰,甘永根,赵玉珂,刘飞飞,何晓蓉. 智能量测开关与智能物联锁具信息交互设计. 中国新技术新产品. 2024(21): 137-139 . 百度学术

    其他类型引用(1)

图(8)  /  表(7)
计量
  • 文章访问数:  559
  • HTML全文浏览量:  83
  • PDF下载量:  235
  • 被引次数: 2
出版历程
  • 收稿日期:  2021-09-22
  • 修回日期:  2022-04-18
  • 网络出版日期:  2023-02-10
  • 刊出日期:  2023-01-31

目录

/

返回文章
返回