[1]XU Lifang,FU Zhijie,MO Hongwei.Tumor recognition in breast ultrasound images based on an improved YOLOv3 algorithm[J].CAAI Transactions on Intelligent Systems,2021,16(1):21-29.[doi:10.11992/tis.202010004]
Copy
CAAI Transactions on Intelligent Systems[ISSN 1673-4785/CN 23-1538/TP] Volume:
16
Number of periods:
2021 1
Page number:
21-29
Column:
学术论文—机器学习
Public date:
2021-01-05
- Title:
-
Tumor recognition in breast ultrasound images based on an improved YOLOv3 algorithm
- Author(s):
-
XU Lifang1; FU Zhijie2; MO Hongwei2
-
1. Engineering Training Center, Harbin Engineering University, Harbin 150001, China;
2. Automation College, Harbin Engineering University, Harbin 150001, China
-
- Keywords:
-
breast cancer; ultrasonography; YOLOv3; SE-Res2Net; downsample block; residual connection; dense connection
- CLC:
-
TP181
- DOI:
-
10.11992/tis.202010004
- Abstract:
-
To improve the efficiency and accuracy of breast cancer diagnoses, a breast ultrasound tumor recognition algorithm based on an improved YOLOV3 algorithm is proposed to assist doctors in breast cancer diagnosis. First, the SE module is integrated into Res2Net to construct Se-Res2Net to replace the original feature extraction network in YOLOv3 to improve the ability of model feature extraction. Then, a new Downsample Block is built to solve the problem of information loss in the downsampling operation of the original model. Finally, to further improve the ability of feature extraction, the residual connection network and dense connection network are combined to construct Res-DenseNet to replace the residual connection mode of the original model. The experimental results show that the above improvements are effective, and the mAP of the improved algorithm is 4.56% higher than that of the original algorithm.