YOLO Object Detector and Inception-V3 Convolutional Neural Network for Improved Brain Tumor Segmentation

被引:6
作者
Shoaib, Muhammad [1 ]
Sayed, Nasir [2 ]
机构
[1] CECOS Univ IT & Emerging Sci, Dept Comp Sci, Peshawar 25000, Khyber Pakhtunk, Pakistan
[2] Islamia Coll Peshawar, Dept Comp Sci, Peshawar 25120, Khyber Pakhtunk, Pakistan
关键词
brain tumor; magnetic resonance imaging; YOLO detector; inception-V3; segmentation; AUTOMATIC SEGMENTATION; IMAGES;
D O I
10.18280/ts.390139
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
For a range of medical analysis applications, the localization of brain tumors and brain tumor segmentation from magnetic resonance imaging (MRI) are challenging yet critical jobs. Many recent studies have included four modalities: i.e., T1, T1c, T2 & FLAIR, it is because every tumor causing area can be detailed examined by each of these brain imaging modalities. Although the BRATS 2018 datasets give impressive segmentation results, the results are still more complex and need more testing and more training. That's why this paper recommends operated pre-processing strategies on a small part of an image except for a full image because that's how an effective and flexible segmented system of brain tumor can be created. In the first phase, an ensemble classification model is developed using different classifiers such as decision tree, SVM, KNN etc. to classify an image into the tumor and non-tumor class by using the strategy of using a small section can completely solve the over-fitting problems and reduces the processing time in a model of YOLO object detector using inceptionv3 CNN features. The second stage is to recommend an efficient and basic Cascade CNN (C-ConvNet/C-CNN), as we deal with a tiny segment of the brain image in each and every slice. In two independent ways, the Cascade-Convolutional Neutral Network model extracts learnable features. On the dataset of BRATS 2018, BRATS 2019 and BRATS 2020, the extensive experimental task has been carried out on the proposed tumor localization framework. The IoU score achieved of three datasets are 97%, 98% and 100%. Other qualitative evaluations & quantitative evaluations are discussed and presented in the manuscript in detail.
引用
收藏
页码:371 / 380
页数:10
相关论文
共 47 条
[1]  
Alpaydin E., 2020, Introduction to machine learning
[2]   Brain tumor detection using statistical and machine learning method [J].
Amin, Javaria ;
Sharif, Muhammad ;
Raza, Mudassar ;
Saba, Tanzila ;
Anjum, Muhammad Almas .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2019, 177 :69-79
[3]  
[Anonymous], 2015, 1511 ARXIV
[4]   SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation [J].
Badrinarayanan, Vijay ;
Kendall, Alex ;
Cipolla, Roberto .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (12) :2481-2495
[5]  
Bakas S., 2018, ARXIV PREPRINT ARXIV
[6]   Fully Automatic Brain Tumor Segmentation using End-To-End Incremental Deep Neural Networks in MRI images [J].
Ben Naceur, Mostefa ;
Saouli, Rachida ;
Akil, Mohamed ;
Kachouri, Rostom .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2018, 166 :39-49
[7]   Deep 3D Convolutional Encoder Networks With Shortcuts for Multiscale Feature Integration Applied to Multiple Sclerosis Lesion Segmentation [J].
Brosch, Tom ;
Tang, Lisa Y. W. ;
Yoo, Youngjin ;
Li, David K. B. ;
Traboulsee, Anthony ;
Tam, Roger .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2016, 35 (05) :1229-1239
[8]   RFDCR: Automated brain lesion segmentation using cascaded random forests with dense conditional random fields [J].
Chen, Gaoxiang ;
Li, Qun ;
Shi, Fuqian ;
Rekik, Islem ;
Pan, Zhifang .
NEUROIMAGE, 2020, 211
[9]   Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation [J].
Chen, Liang-Chieh ;
Zhu, Yukun ;
Papandreou, George ;
Schroff, Florian ;
Adam, Hartwig .
COMPUTER VISION - ECCV 2018, PT VII, 2018, 11211 :833-851
[10]   DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs [J].
Chen, Liang-Chieh ;
Papandreou, George ;
Kokkinos, Iasonas ;
Murphy, Kevin ;
Yuille, Alan L. .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2018, 40 (04) :834-848