This is the source code for the paper, "BGF-YOLO: Enhanced YOLOv8 with Multiscale Attentional Feature Fusion for Brain Tumor Detection", early 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.
The Bilevel routing attention, Generalized feature pyramid networks, and Fourth detecting head You Only Look Once (BGF-YOLO) 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/.
Install requirements.txt in a Python >= 3.8 environment, including PyTorch >= 1.8.
pip install -r requirements.txt
python yolo/bgf/detect/train.py
python yolo/bgf/detect/predict.py
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.
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. 15008, in press (000–000). Springer, Cham (2024). https://doi.org/10.1007/978-3-031-72110-6_4
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 VIII (eds. Xxxxx, X. et al.) in press (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, in press (pp. 000–000).
NOTE: City of Conf., Abbrev. State, Country, Month & Day(s) are optional. -
Elsevier Numbered 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, Germany, 2024, in press (pp. 000–000).
NOTE: Day(s) Month Year, City, Abbrev. State, Country of Conference, Publiser, and Place of Publication are optional. -
Harvard (Name–Date) Style
Kang, M., Ting, C.-M., Ting, F.F., Phan, R.C.-W., 2024. 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, Germany, in press (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. 15008, 000--000. Springer, Cham (2024). {\UrlFont https://doi.org/10.1007/978-3-031-72110-6\_4}
\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 VIII",
series = "Lecture Notes in Computer Science (LNCS)",
volume = "15008",
pages = "000--000",
publisher = "Springer",
address = "Cham",
year = "2024",
doi= "10.1007/978-3-031-72110-6_4",
url = "https://doi.org/10.1007/978-3-031-72110-6_4"
}
@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 {}.
BGF-YOLO is released under the GNU Affero General Public License v3.0 (AGPL-3.0). Please see the LICENSE file for more information.
Many utility codes of our project base on the codes of Ultralytics YOLOv8, GiraffeDet, DAMO-YOLO, and BiFormer repositories.