Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Muhammad Rizwan Munawar 2024-04-13 20:22:52 +05:00 committed by GitHub
parent 42416bc608
commit 0f4a4fdf5f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
19 changed files with 30 additions and 4 deletions

View file

@ -6,6 +6,7 @@ import torch.nn.functional as F
from ultralytics.utils.loss import FocalLoss, VarifocalLoss
from ultralytics.utils.metrics import bbox_iou
from .ops import HungarianMatcher