Academic Journal of Computing & Information Science, 2025, 8(3); doi: 10.25236/AJCIS.2025.080308.
Song Wei1, Wang Shengjie2
1The First Affiliated Hospital of Anhui Medical University, Anhui Medical University, Hefei, China
2International Sakharov Environmental Institute, Belarusian State University, Minsk, Belarus
With the widespread application of computed tomography (CT) imaging technology, the detection of aortic dilation has become an important clinical diagnostic tool. This study proposes an improved model based on YOLOv5 for the automatic detection of aortic dilation in CT images. To address the insufficient feature extraction of low-contrast structures and the multi-scale fusion issues in the original YOLOv5 model when applied to medical imaging, ResNet50 is introduced as the backbone network. Its deep residual structure enhances the recognition of vascular walls and calcified plaques. Additionally, a dynamic weighted bidirectional feature pyramid network (BiFPN) is used to replace PANet, achieving adaptive feature fusion of multi-scale vascular structures. Experiments conducted on a dataset consisting of 9,856 CT images show that the improved model achieves an [email protected] of 88.79%, a 6.62% increase compared to the baseline model, while maintaining real-time performance. Notably, for small lesions (diameter < 10mm), the recall rate improved by 19.3%. Although the computational complexity of the model increased, it still meets the requirements for real-time clinical detection. The experimental results validate the effectiveness of the improved model in detecting aortic dilation, providing new optimization ideas and application potential for object detection in medical imaging.
YOLOv5; Aortic Dilation; CT Imaging; ResNet50; BiFPN
Song Wei, Wang Shengjie. Aortic Aneurysm Detection in CT Medical Images Based on YOLOv5. Academic Journal of Computing & Information Science(2025), Vol. 8, Issue 3: 56-63. https://doi.org/10.25236/AJCIS.2025.080308.
[1] Jocher G, Stoken A, Borovec J, et al. ultralytics/yolov5: v3. 0[J]. Zenodo, 2020.
[2] Elpeltagy M, Sallam H. Automatic prediction of COVID− 19 from chest images using modified ResNet50[J]. Multimedia tools and applications, 2021, 80(17): 26451-26463.
[3] Chen J, Mai H S, Luo L, et al. Effective feature fusion network in BIFPN for small object detection[C]//2021 IEEE international conference on image processing (ICIP). IEEE, 2021: 699-703.