A Two-Tier Framework Based on GoogLeNet and YOLOv3 Models for Tumor Detection in MRI

Farman Ali, Sadia Khan, Arbab Waseem Abbas, Babar Shah, Tariq Hussain, Dongho Song, Shaker EI-Sappagh, Jaiteg Singh

Research output: Contribution to journalArticlepeer-review

4 Scopus citations

Abstract

Medical Image Analysis (MIA) is one of the active research areas in computer vision, where brain tumor detection is the most investigated domain among researchers due to its deadly nature. Brain tumor detection in magnetic resonance imaging (MRI) assists radiologists for better analysis about the exact size and location of the tumor. However, the existing systems may not efficiently classify the human brain tumors with significantly higher accuracies. In addition, smart and easily implementable approaches are unavailable in 2D and 3D medical images, which is the main problem in detecting the tumor. In this paper, we investigate various deep learning models for the detection and localization of the tumor in MRI. A novel two-tier framework is proposed where the first tire classifies normal and tumor MRI followed by tumor regions localization in the second tire. Furthermore, in this paper, we introduce a well-annotated dataset comprised of tumor and normal images. The experimental results demonstrate the effectiveness of the proposed framework by achieving 97% accuracy using GoogLeNet on the proposed dataset for classification and 83% for localization tasks after fine-tuning the pre-trained you only look once (YOLO) v3 model.

Original languageEnglish
Pages (from-to)73-92
Number of pages20
JournalComputers, Materials and Continua
Volume72
Issue number1
DOIs
StatePublished - 2022

Keywords

  • GoogLeNet
  • Image classification
  • MRI
  • Tumor localization
  • YOLOv3

Fingerprint

Dive into the research topics of 'A Two-Tier Framework Based on GoogLeNet and YOLOv3 Models for Tumor Detection in MRI'. Together they form a unique fingerprint.

Cite this