search

mkang315 / BGF-YOLO

[MICCAI'24] Official implementation of "BGF-YOLO: Enhanced YOLOv8 with Multiscale Attentional Feature Fusion for Brain Tumor Detection".
GNU Affero General Public License v3.0
70 stars 11 forks source link

Asking about wiou #6

Open supriamir opened 7 months ago

supriamir commented 7 months ago

Hi @mkang315

I would like to inquire about the wIoU loss function. If I want to implement wIoU based on your code, would any settings need to be changed?

I already follow your notes. But I just wonder how to running it.

In yolov5 to use other loss function, we need to change CIOU=False.

Thank you

mkang315 commented 7 months ago

To use \BGF-YOLO\yolo\utils\wiou.py, only changing CIoU=True to CIoU=False in the loss.py. \BGF-YOLO\yolo\utils\loss.py iou = bbox_iou(pred_bboxes[fg_mask], target_bboxes[fg_mask], xywh=False, CIoU=True)

No need to change another code.