Construction of artificial intelligence models for multi-category lesion detection in small bowel capsule endoscopy based on various YOLO neural networks
10.3969/j.issn.1005-202X.2025.05.021
- VernacularTitle:基于不同YOLO神经网络建立小肠胶囊内镜多类别病变自动检测的人工智能模型
- Author:
Jian CHEN
1
;
Ganhong WANG
;
Jianjun DAI
;
Kaijian XIA
;
Xiaodan XU
;
Ying SUN
Author Information
1. 常熟市第一人民医院消化内科,江苏常熟 215500
- Publication Type:Journal Article
- Keywords:
small bowel lesion;
YOLOv10;
artificial intelligence;
small bowel capsule endoscopy;
object detection
- From:
Chinese Journal of Medical Physics
2025;42(5):693-700
- CountryChina
- Language:Chinese
-
Abstract:
Objective To construct YOLOv10 based artificial intelligence(AI)models for the automatic detection in small bowel capsule endoscopy(SBCE)images.Methods SBCE data from two centers was collected,including 23 115 images and 35 412 annotated labels covering 11 categories of small bowel lesions.The images were annotated using the LabelMe tool and converted into the YOLO format required for deep learning model development.The pre-trained YOLOv10 and YOLOv8 models were used for transfer learning training on the constructed dataset.Model performance was comprehensively evaluated using metrics such as precision,accuracy,sensitivity,specificity,false-positive rate,and detection speed.Finally,the models were deployed on local computers for real-time detection of SBCE images and videos.Results Six different versions of YOLO object detection models were developed,namely YOLOv8n,YOLOv8s,YOLOv8m,YOLOv10n,YOLOv10s,and YOLOv10m.On the validation set,YOLOv10s model achieved the best mAP50(0.795);although its inference latency was not the fastest(4.803 ms/img),it met the requirements for clinical application.On the test set,YOLOv10s performed well,with an accuracy of 92.69%,a sensitivity of 89.23%,and a false-positive rate of 4.78%.Especially,in category-specific inference,the highest sensitivity was for"bleeding"at 96.41%,while the lowest was for"narrowing"at 82.29%.Conclusion The model constructed based on YOLOv10 neural network can rapidly and accurately detect and classify various small bowel lesions,exhibiting significant clinical application potential.
