YOLOv7 for brain tumour detection using morphological transfer learning model

被引:1
|
作者
Pandey, Sanat Kumar [1 ]
Bhandari, Ashish Kumar [1 ]
机构
[1] Department of Electronics and Communication Engineering, National Institute of Technology Patna, Patna,800005, India
关键词
Contrastive Learning - Deep learning - Image enhancement - Nuclear magnetic resonance - Transfer learning;
D O I
10.1007/s00521-024-10246-7
中图分类号
学科分类号
摘要
An accurate diagnosis of a brain tumour in its early stages is required to improve the possibility of survival for cancer patients. Due to the structural complexity of the brain, it has become very difficult and tedious for neurologists and radiologists to diagnose brain tumours in the initial stages with the help of various common manual approaches to tumour diagnosis. To improve the performance of the diagnosis, some computer-aided diagnosis-based systems are developed with the concepts of artificial intelligence. In this proposed manuscript, we analyse various computer-aided design (CAD)-based approaches and design a modern approach with ideas of transfer learning over deep learning on magnetic resonance imaging (MRI). In this study, we apply a transfer learning approach with the object detection model YOLO (You Only Look Once) and analyse the MRI dataset with the various modified versions of YOLO. After the analysis, we propose an object detection model based on the modified YOLOv7 with a morphological filtering approach to reach an efficient and accurate diagnosis. To enhance the performance accuracy of this suggested model, we also analyse the various versions of YOLOv7 models and find that the proposed model having the YOLOv7-E6E object detection technique gives the optimum value of performance indicators as precision, recall, F1, and mAP@50 as 1, 0.92, 0.958333, and 0.974, respectively. The value of mAP@50 improves to 0.992 by introducing a morphological filtering approach before the object detection technique. During the complete analysis of the suggested model, we use the BraTS 2021 dataset. The BraTS 2021 dataset has brain MR images from the RSNA-MICCAI brain tumour radiogenetic competition, and the complete dataset is labelled using the online tool MakeSense AI. © The Author(s), under exclusive licence to Springer-Verlag London Ltd., part of Springer Nature 2024.
引用
收藏
页码:20321 / 20340
页数:19
相关论文
共 50 条
  • [1] A Trash Detection Model Based on YOLOv7
    Liang, Hu
    Xu, Chao
    He, Tao
    PROCEEDINGS OF 2024 INTERNATIONAL CONFERENCE ON COMPUTER AND MULTIMEDIA TECHNOLOGY, ICCMT 2024, 2024, : 300 - 303
  • [2] Morphological transfer learning based brain tumor detection using YOLOv5
    Pandey, Sanat Kumar
    Bhandari, Ashish Kumar
    MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 83 (16) : 49343 - 49366
  • [3] Morphological transfer learning based brain tumor detection using YOLOv5
    Sanat Kumar Pandey
    Ashish Kumar Bhandari
    Multimedia Tools and Applications, 2024, 83 : 49343 - 49366
  • [4] Improved YOLOv7 model for insulator defect detection
    Wang, Zhenyue
    Yuan, Guowu
    Zhou, Hao
    Ma, Yi
    Ma, Yutang
    Chen, Dong
    ELECTRONIC RESEARCH ARCHIVE, 2024, 32 (04): : 2880 - 2896
  • [5] CCG-YOLOv7: A Wood Defect Detection Model for Small Targets Using Improved YOLOv7
    Cui, Wenqi
    Li, Zhenye
    Duanmu, Anning
    Xue, Sheng
    Guo, Yiren
    Ni, Chao
    Zhu, Tingting
    Zhang, Yajun
    IEEE ACCESS, 2024, 12 : 10575 - 10585
  • [6] Active learning based fire detection method for YOLOv7
    Lu, Yangzhengyu
    Sang, Qingbing
    PROCEEDINGS OF 2024 3RD INTERNATIONAL CONFERENCE ON CYBER SECURITY, ARTIFICIAL INTELLIGENCE AND DIGITAL ECONOMY, CSAIDE 2024, 2024, : 320 - 324
  • [7] Object Detection with YOLOv7 Model on Smart Mobile Devices
    Karadag, Batuhan
    Ari, Ali
    JOURNAL OF POLYTECHNIC-POLITEKNIK DERGISI, 2023, 26 (03): : 1207 - 1214
  • [8] Automatic Acne Detection Model Based on Improved YOLOv7
    Zhang, Delong
    Jin, Chunyang
    Zhang, Zhidong
    Cao, Xiyuan
    Xue, Chenyang
    IEEE ACCESS, 2024, 12 : 194390 - 194398
  • [9] Enhancing Caries Detection in Bitewing Radiographs Using YOLOv7
    Panyarak, Wannakamon
    Wantanajittikul, Kittichai
    Charuakkra, Arnon
    Prapayasatok, Sangsom
    Suttapak, Wattanapong
    JOURNAL OF DIGITAL IMAGING, 2023, 36 (06) : 2635 - 2647
  • [10] Enhancing Caries Detection in Bitewing Radiographs Using YOLOv7
    Wannakamon Panyarak
    Kittichai Wantanajittikul
    Arnon Charuakkra
    Sangsom Prapayasatok
    Wattanapong Suttapak
    Journal of Digital Imaging, 2023, 36 : 2635 - 2647