Detection and classification of COVID-19 by using faster R-CNN and mask R-CNN on CT images

被引:43
作者
Sahin, M. Emin [1 ]
Ulutas, Hasan [1 ]
Yuce, Esra [1 ]
Erkoc, Mustafa Fatih [2 ]
机构
[1] Yozgat Bozok Univ, Fac Engn & Architecture, Dept Comp Engn, Yozgat, Turkiye
[2] Yozgat Bozok Univ, Fac Med, Dept Radiol, Yozgat, Turkiye
关键词
Classification; COVID-19; CT images; Faster R-CNN; Mask R-CNN;
D O I
10.1007/s00521-023-08450-y
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The coronavirus (COVID-19) pandemic has a devastating impact on people's daily lives and healthcare systems. The rapid spread of this virus should be stopped by early detection of infected patients through efficient screening. Artificial intelligence techniques are used for accurate disease detection in computed tomography (CT) images. This article aims to develop a process that can accurately diagnose COVID-19 using deep learning techniques on CT images. Using CT images collected from Yozgat Bozok University, the presented method begins with the creation of an original dataset, which includes 4000 CT images. The faster R-CNN and mask R-CNN methods are presented for this purpose in order to train and test the dataset to categorize patients with COVID-19 and pneumonia infections. In this study, the results are compared using VGG-16 for faster R-CNN model and ResNet-50 and ResNet-101 backbones for mask R-CNN. The faster R-CNN model used in the study has an accuracy rate of 93.86%, and the ROI (region of interest) classification loss is 0.061 per ROI. At the conclusion of the final training, the mask R-CNN model generates mAP (mean average precision) values for ResNet-50 and ResNet-101, respectively, of 97.72% and 95.65%. The results for five folds are obtained by applying the cross-validation to the methods used. With training, our model performs better than the industry standard baselines and can help with automated COVID-19 severity quantification in CT images.
引用
收藏
页码:13597 / 13611
页数:15
相关论文
共 39 条
[1]  
Abdulla W., 2017, GITHUB REPOSITORY
[2]   Application of deep learning technique to manage COVID-19 in routine clinical practice using CT images: Results of 10 convolutional neural networks [J].
Ardakani, Ali Abbasian ;
Kanafi, Alireza Rajabzadeh ;
Acharya, U. Rajendra ;
Khadem, Nazanin ;
Mohammadi, Afshin .
COMPUTERS IN BIOLOGY AND MEDICINE, 2020, 121
[3]  
Berrar D., 2019, Encyclopedia of Bioinformatics and Computational Biology, Volume, VI, P542, DOI [10.1016/B978-0-12-809633-8.20349-X, DOI 10.1016/B978-0-12-809633-8.20349-X]
[4]   Gastric Cancer Diagnosis with Mask R-CNN [J].
Cao, Guitao ;
Song, Wenli ;
Zhao, Zhenwei .
2019 11TH INTERNATIONAL CONFERENCE ON INTELLIGENT HUMAN-MACHINE SYSTEMS AND CYBERNETICS (IHMSC 2019), VOL 1, 2019, :60-63
[5]   Detection and classification the breast tumors using mask R-CNN on sonograms [J].
Chiao, Jui-Ying ;
Chen, Kuan-Yung ;
Liao, Ken Ying-Kai ;
Hsieh, Po-Hsin ;
Zhang, Geoffrey ;
Huang, Tzung-Chi .
MEDICINE, 2019, 98 (19)
[6]  
Nguyen DH, 2018, PROCEEDINGS OF THE 2018 5TH ASIAN CONFERENCE ON DEFENSE TECHNOLOGY (ACDT 2018), P14, DOI 10.1109/ACDT.2018.8593130
[7]   COVID-19 CT image recognition algorithm based on transformer and CNN [J].
Fan, Xiaole ;
Feng, Xiufang ;
Dong, Yunyun ;
Hou, Huichao .
DISPLAYS, 2022, 72
[8]   Rich feature hierarchies for accurate object detection and semantic segmentation [J].
Girshick, Ross ;
Donahue, Jeff ;
Darrell, Trevor ;
Malik, Jitendra .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :580-587
[9]  
He KM, 2020, IEEE T PATTERN ANAL, V42, P386, DOI [10.1109/TPAMI.2018.2844175, 10.1109/ICCV.2017.322]
[10]  
He KM, 2014, LECT NOTES COMPUT SC, V8691, P346, DOI [arXiv:1406.4729, 10.1007/978-3-319-10578-9_23]