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

VernacularTitle:基于改进YOLOv7的肝囊型包虫病超声图像小病灶检测
Author: Miwueryiti HAILATI ¹ ; Renaguli AIHEMAITINIYAZI ; Kadiliya KUERBAN ; Chuanbo YAN
Author Information

1. 新疆医科大学公共卫生学院,新疆乌鲁木齐 830011
Keywords: cystic echinococcosis; deep learning; object detection; YOLOv7; ECIoU; GhostNet
From: Chinese Journal of Medical Physics 2024;41(3):299-308
CountryChina
Language:Chinese
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.