Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Asking about wiou #6

Open
supriamir opened this issue Jan 27, 2024 · 1 comment
Open

Asking about wiou #6

supriamir opened this issue Jan 27, 2024 · 1 comment

Comments

@supriamir
Copy link

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
Copy link
Owner

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants