Brain Tumor Localization with ViT and a Loss Function Based on Complete Intersection Over Union
摘要
This paper introduces a novel methodology that integrates a Vision Transformer with a loss function based on Complete Intersection over Union (CIOU) to achieve precise object localization and bounding box regression. Experimental results on a brain tumor dataset demonstrate that our approach improves the performance of ViT compared to alternative loss functions such as Mean Square Error (MSE), IoU, GIoU, and DGIoU.