范博, 吴俊, 孙亮, 颜光前, 张兆民, 王强, 夏苠芩. 基于改进YOLOv3和迁移学习的轻量型补片目标检测[J]. 云南大学学报(自然科学版), 2022, 44(3): 471-479. doi: 10.7540/j.ynu.20210354
引用本文: 范博, 吴俊, 孙亮, 颜光前, 张兆民, 王强, 夏苠芩. 基于改进YOLOv3和迁移学习的轻量型补片目标检测[J]. 云南大学学报(自然科学版), 2022, 44(3): 471-479. doi: 10.7540/j.ynu.20210354
FAN Bo, WU Jun, SUN Liang, YAN Guang-qian, ZHANG Zhao-min, WANG Qiang, XIA Min-qin. Lightweight mesh target detection based on improved YOLOv3 and transfer learning[J]. Journal of Yunnan University: Natural Sciences Edition, 2022, 44(3): 471-479. DOI: 10.7540/j.ynu.20210354
Citation: FAN Bo, WU Jun, SUN Liang, YAN Guang-qian, ZHANG Zhao-min, WANG Qiang, XIA Min-qin. Lightweight mesh target detection based on improved YOLOv3 and transfer learning[J]. Journal of Yunnan University: Natural Sciences Edition, 2022, 44(3): 471-479. DOI: 10.7540/j.ynu.20210354

基于改进YOLOv3和迁移学习的轻量型补片目标检测

Lightweight mesh target detection based on improved YOLOv3 and transfer learning

  • 摘要: 自动三维乳腺超声(Automated 3-D Breast Ultrasound,ABUS)克服传统超声的缺陷,成功应用于对腹壁疝轻量型补片的检查. 但人工检阅ABUS超声图像耗时费力,且极易出现漏诊等问题. 因此,文章提出一种基于改进YOLOv3和迁移学习的目标检测算法以辅助医生提高审阅速度和准确性. 基于原有的YOLOv3模型,在检测层前增加空间金字塔池化 (Spatial Pyramid Pooling,SPP)模块实现局部特征与全局特征的融合,丰富特征图的表达,解决了检测图像中小目标难以检测的问题;在网络训练中,采用迁移学习的策略进行训练网络以克服轻量型补片图像数据集有限的问题,提升网络的鲁棒性减少过拟合产生. 实验结果表明,YOLOv3-SPP算法结合迁移学习训练方式,其平均精度均值(mean Average Precision,mAP)达到90.15%,图像检测速度为33.2 f·s−1,可有效辅助医生提高审阅效率.

     

    Abstract: Automatic 3D Breast Ultrasound (ABUS) overcomes the shortcomings of traditional ultrasound and is successfully applied to the examination of lightweight mesh for abdominal hernia. However, manual inspection of ABUS ultrasound images is time-consuming and prone to missed diagnosis. Therefore, this paper proposes a target detection algorithm based on improved YOLOv3 and transfer learning to assist doctors to improve review speed and accuracy. Based on the original YOLOv3 model, a Spatial Pyramid Pooling (SPP) module is added in front of the detection layer to realize the fusion of local features and global features, which enriches the expression ability of the feature graph and solves the problem that it is difficult to detect small targets in the detection image. In the network training, the transfer learning strategy is used to train the network to overcome the problem of limited light mesh image data set and improve the robustness of the network to reduce over-fitting. The experimental results show that the mean Average Precision (mAP) of YOLOv3-SPP algorithm combined with transfer learning training method is 90.15%, and the image detection speed is 33.2 f·s−1, which can effectively assist doctors to improve review efficiency.

     

/

返回文章
返回