Improved polyp detection from colonoscopy images using finetuned YOLO-v5

被引:9
作者
Ghose, Priyanka [1 ]
Ghose, Arpan [2 ]
Sadhukhan, Deboleena [3 ]
Pal, Saurabh [4 ]
Mitra, Madhuchanda [4 ]
机构
[1] GCETTS Govt Coll Engn & Text Technol, Serampore, India
[2] Capgemini Technol Serv India Ltd, Kolkata, India
[3] Inst Langevin Paris, Paris, France
[4] Univ Calcutta, Fac Council Postgrad Studies Engn & Technol, Kolkata, India
关键词
Colonoscopy image; Image processing; Data augmentation; Deep learning; Object detection; YOLO-v5;
D O I
10.1007/s11042-023-17138-3
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Object detection plays an important role to accelerate the medical diagnosis and automatic polyp detection from colonoscopy images is one of the prominent examples. Visual examination is an error prone and time-consuming process to determine the shape, size, and location of polyps in colonoscopy images. Deep learning models are extensively accepted in the field of object detection. YOLO-v5 is an object detection model which is powered by deep learning technique. In this study we suggest a finetuned YOLO-v5 model for polyp detection from colonoscopy images. Effective data augmentation is also done to improvise the performance of the model. To quantify the efficiency of our solution against other works, two aspects have been fundamentally considered - qualitative performance and practical reliability, which have been achieved successfully. A detailed comparison study has been shared to justify the performance of our proposed method against other deep learning techniques - R-CNN, Faster RCNN and YOLO-v4. Our proposed solution deliberated high value of performance metrics which are comparatively better than other solutions.
引用
收藏
页码:42929 / 42954
页数:26
相关论文
共 49 条
[1]  
Agrawal P, 2014, LECT NOTES COMPUT SC, V8695, P329, DOI 10.1007/978-3-319-10584-0_22
[2]   Human Detection in Aerial Thermal Images Using Faster R-CNN and SSD Algorithms [J].
Akshatha, K. R. ;
Karunakar, A. Kotegar ;
Shenoy, Satish B. ;
Pai, Abhilash K. ;
Nagaraj, Nikhil Hunjanal ;
Rohatgi, Sambhav Singh .
ELECTRONICS, 2022, 11 (07)
[3]   Towards Real-Time Polyp Detection in Colonoscopy Videos: Adapting Still Frame-Based Methodologies for Video Sequences Analysis [J].
Angermann, Quentin ;
Bernal, Jorge ;
Sanchez-Montes, Cristina ;
Hammami, Maroua ;
Fernandez-Esparrach, Gloria ;
Dray, Xavier ;
Romain, Olivier ;
Javier Sanchez, F. ;
Histace, Aymeric .
COMPUTER ASSISTED AND ROBOTIC ENDOSCOPY AND CLINICAL IMAGE-BASED PROCEDURES, 2017, 10550 :29-41
[4]  
Bardhi O, 2017, IEEE INT SYMP SIGNAL, P445, DOI 10.1109/ISSPIT.2017.8388684
[5]  
Bernal J., 2018, P 32 CARS C
[6]  
Bochkovskiy A, 2020, Arxiv, DOI arXiv:2004.10934
[7]   Gastric polyp detection in gastroscopic images using deep neural network [J].
Cao, Chanting ;
Wang, Ruilin ;
Yu, Yao ;
Zhang, Hui ;
Yu, Ying ;
Sun, Changyin .
PLOS ONE, 2021, 16 (04)
[8]   Vehicle Detection Using Different Deep Learning Algorithms from Image Sequence [J].
Cepni, Sumeyye ;
Atik, Muhammed Enes ;
Duran, Zaide .
BALTIC JOURNAL OF MODERN COMPUTING, 2020, 8 (02) :347-358
[9]   A self-attention based faster R-CNN for polyp detection from colonoscopy images [J].
Chen, Bo-Lun ;
Wan, Jing-Jing ;
Chen, Tai-Yue ;
Yu, Yong-Tao ;
Ji, Min .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2021, 70
[10]   Improving deep learning-based polyp detection using feature extraction and data augmentation [J].
Chou, Yung-Chien ;
Chen, Chao-Chun .
MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 82 (11) :16817-16837