/BGF-YOLO

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

Primary LanguagePythonGNU Affero General Public License v3.0AGPL-3.0

Official BGF-YOLO

This is the source code for the paper, "Bgf-yolo: Enhanced YOLOv8 with Multiscale Attentional Feature Fusion for Brain Tumor Detection", of which I am the first author.

Model

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

Training

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

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..

Suggested Citation

Our manuscript has been uploaded on arXiv. Please cite our paper if you use code from this repository:

Plain Text

  • IEEE 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," arXiv:2309.12585 [cs.CV], Jun. 2023.

  • Nature 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. Preprint at https://arxiv.org/abs/2309.12585 (2023).

  • Springer 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. arXiv preprint arXiv:2309.12585 (2023)

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.