Capsule Endoscopy Lesion Area Detection Based on an Improved YOLOv5 Algorithm

Ji-hui TU , Ya-nan XIAO , Xue-kui BU , Qing ZHANG , Jie LI

Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (7) : 2821 -2828.

PDF
Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (7) : 2821-2828 DOI: 10.12404/j.issn.1671-1815.2304614
research-article
Capsule Endoscopy Lesion Area Detection Based on an Improved YOLOv5 Algorithm
Author information +
History +
PDF

Abstract

Since the existing capsule endoscopy detection model has the problem of single lesion detection and low efficiency, a novel lesion region detection method of capsule endoscopic based on the improved YOLOv5 was proposed. The original YOLOv5 had been improved in the following three aspects. Firstly, the convolutional block attention module (CBAM) was incorporated into the Backbone of YOLOv5, which was used to enhance the presentation ability of the target feature. Secondly, a detection head was added to the Head of YOLOv5, which was used to enhance the detection ability of the small targets. Finally, the GIOU loss function of YOLOv5 was replaced with the CIOU loss function, which leaded to quicker convergence during training. The proposed method was tested on the capsule endoscopic image data provided by the First Clinical Medical College of Yangtze University. It's accuracy rate reached 93.6%, the recall rate reached 94.3%, the detection rate reached 97.2% on mAP@0.5 and a detection speed of 0.027 2 s per frame for lesion area detection on the dataset. The experimental results show that the proposed method is effective, flexible, and robust, which can meet the practical needs of clinical medical diagnosis.

Cite this article

Download citation ▾
Ji-hui TU, Ya-nan XIAO, Xue-kui BU, Qing ZHANG, Jie LI. Capsule Endoscopy Lesion Area Detection Based on an Improved YOLOv5 Algorithm. Science Technology and Engineering, 2024, 24(7): 2821-2828 DOI:10.12404/j.issn.1671-1815.2304614

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (2923KB)

19

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉