Abstract

Info

Title:: Small lesion detection in ultrasound images of hepatic cystic echinococcosis based on improved YOLOv7

Author(s):: HAILATI Miwueryiti1; AIHEMAITINIYAZI Renaguli1; KUERBAN Kadiliya1; YAN Chuanbo2; 1. College of Public Health, Xinjiang Medical University, Urumqi 830011, China 2. College of Medical Engineering Technology, Xinjiang Medical University, Urumqi 830011, China

Keywords:: Keywords: cystic echinococcosis deep learning object detection YOLOv7 ECIoU GhostNet

Abstract:: Abstract: Objective To propose a novel algorithm model based on YOLOv7 for detecting small lesions in ultrasound images of hepatic cystic echinococcosis. Methods The original feature extraction backbone was replaced with a lightweight feature extraction backbone network GhostNet for reducing the quantity of model parameters. To address the problem of low detection accuracy when the evaluation index CIoU of YOLOv7 was used as a loss function, ECIoU was substituting for CIoU, which further improved the model detection accuracy. Results The model was trained on a self-built dataset of small lesion ultrasound images of hepatic cystic echinococcosis. The results showed that the improved model had a size of 59.4 G and a detection accuracy of 88.1% for mAP@0.5, outperforming the original model and surpassing other mainstream detection methods. Conclusion The proposed model can detect and classify the location and category of lesions in ultrasound images of hepatic cystic echinococcosis more efficiently.