mkang315 / BGF-YOLO

Official implementation of "BGF-YOLO: Enhanced YOLOv8 with Multiscale Attentional Feature Fusion for Brain Tumor Detection".

Official BGF-YOLO

This is the source code for the paper, "BGF-YOLO: Enhanced YOLOv8 with Multiscale Attentional Feature Fusion for Brain Tumor Detection", accepted by the 27th International Conference on Medical Image Computing and Computer Assisted Intervention (MICCAI 2024), of which I am the first author. The paper is available to download from arXiv.

Model

The model configuration (i.e., network construction) file is bgf-yolo.yaml in the directory ./models/bgf.

The hyperparameter setting file is default.yaml in the directory ./yolo/cfg/.

Installation

Install requirements.txt in a Python >= 3.8 environment, including PyTorch >= 1.8.

pip install -r requirements.txt  # install

Training CLI

python yolo/bgf/detect/train.py

Testing CLI

python yolo/bgf/detect/predict.py

Evaluation

We trained and evaluated BGF-YOLO on the dataset Br35H :: Brain Tumor Detection 2020. The .txt format annotations in the file dataset-Br35H.zip are coverted from original json format.

Suggested Citation

Please cite our paper if you use code from this repository:

Plain Text

• Springer Reference Style
  Kang, M., Ting, C.-M., Ting, F.F., Phan, R.C.-W.: BGF-YOLO: enhanced YOLOv8 with multiscale attentional feature fusion for brain tumor detection. In: Xxxxx, X., et al. (eds.) MICCAI 2024. LNCS, vol. 00000, 000–010. Springer, Cham (2024). https://doi.org/10.1007/978-3-000-00000-0_00
  ^{NOTE: MICCAI conference proceedings are part of the book series LNCS in which Springer's format for bibliographical references is strictly enforced. This is important, for instance, when citing previous MICCAI proceedings. LNCS stands for Lecture Notes in Computer Science.}

• Nature Reference Style
  Kang, M., Ting, C.-M., Ting, F. F. & Phan, R. C.-W. BGF-YOLO: enhanced YOLOv8 with multiscale attentional feature fusion for brain tumor detection. In Medical Image Computing and Computer-Assisted Intervention – MICCAI 2024: 27th International Conference, Marrakesh, Morocco, October 6–10, 2024, Proceedings, Part IV (eds. Xxxxx, X. et al.) 000–000 (Springer, 2024).
  

• IEEE Reference Style
  M. Kang, C.-M. Ting, F. F. Ting, and R. C.-W. Phan, "Bgf-yolo: Enhanced yolov8 with multiscale attentional feature fusion for brain tumor detection," in Proc. Int. Conf. Med. Image Comput. Comput. Assist. Interv. (MICCAI), Marrakesh, Morocco, Oct. 6–10, 2024, pp. 000–000.
  ^{NOTE: City of Conf., Abbrev. State, Country, Month & day(s) are optional.}

• Elsevier Reference Style
  M. Kang, C.-M. Ting, F.F. Ting, R.C.-W. Phan, BGF-YOLO: enhanced YOLOv8 with multiscale attentional feature fusion for brain tumor detection, in: X. Xxxx, X. Xxxx, X. Xxxx, X. Xxxx, X. Xxxx, X. Xxxx, et al. (Eds.), Proceedings of the 27th International Conference on Medical Image Computing and Computer Assisted Intervention (MICCAI), 6–10 October 2024, Marrakesh, Morocco, Springer, Cham, 2024, pp. 000–000.
  ^{NOTE: Day(s) Month Year, City, Abbrev. State, Country of Conference, Publiser, and Place of Publication are optional.}

BibTeX Format

\begin{thebibliography}{1}
\bibitem{Kang24Bgfyolo} Kang, M., Ting, C.-M., Ting, F.F., Phan, R.C.-W.: BGF-YOLO: enhanced YOLOv8 with multiscale attentional feature fusion for brain tumor detection. In: Xxxx, X., et al. (eds.) MICCAI 2024. LNCS, vol. 00000, 000--000. Springer, Cham (2024). {\UrlFont https://doi.org/10.1007/978-3-000-00000-0\_00}
\end{thebibliography}

@inproceedings{Kang24Bgfyolo,
  author = "Kang, Ming and Ting, Chee-Ming and Ting, Fung Fung and Phan, Rapha{\"e}l C.-W.",
  title = "BGF-YOLO: enhanced YOLOv8 with multiscale attentional feature fusion for brain tumor detection",
  editor = "Xxxxx, Xxxx and et al.",
  booktitle = "Medical Image Computing and Computer-Assisted Intervention – MICCAI 2024: 27th International Conference, Marrakesh, Morocco, October 6--10, 2024, Proceedings, Part XX",
  series = "Lecture Notes in Computer Science (LNCS)",
  volume = "00000",
  pages = "000--000",
  publisher = "Springer",
  address = "Cham",
  year = "2024",
  doi= "10.1007/978-3-031-00000-0\_00",
  url = "https://doi.org/10.1007/978-3-000-00000-0\_00"
}

@inproceedings{Kang24Bgfyolo,
  author = "Ming Kang and Chee-Ming Ting and Fung Fung Ting and Rapha{\"e}l C.-W. Phan",
  title = "Bgf-yolo: Enhanced yolov8 with multiscale attentional feature fusion for brain tumor detection",
  booktitle = "Proc. Int. Conf. Med. Image Comput. Comput. Assist. Interv. (MICCAI)",
  address = "Marrakesh, Morocco, Oct. 6--10",
  pages = "000--000",
  year = "2024",
}

^{NOTE: Please remove some optional BibTeX fields, for example, series, volume, address, url and so on, while the LaTeX compiler produces an error. Author names may be manually modified if not automatically abbreviated by the compiler under the control of the .bst file if applicable which defines bibliography/reference style. Kang24Bgfyolo could be b1, bib1, or ref1 when references appear in numbered style in which they are cited. The quotation mark pair "" in the field could be replaced by the brace {}.}

License

BGF-YOLO is released under the GNU Affero General Public License v3.0 (AGPL-3.0). Please see the LICENSE file for more information.

Copyright Notice

Many utility codes of our project base on the codes of Ultralytics YOLOv8, GiraffeDet, DAMO-YOLO, and BiFormer repositories.