update

yolo/models/yolov5_af/matcher.py

@@ -1,8 +1,3 @@
-# ---------------------------------------------------------------------
-# Copyright (c) Megvii Inc. All rights reserved.
-# ---------------------------------------------------------------------
-
-
 import torch
 import torch.nn.functional as F
 from utils.box_ops import *

yolo/models/yolov8/matcher.py

@@ -6,6 +6,9 @@ from utils.box_ops import bbox_iou
 
 # -------------------------- Task Aligned Assigner --------------------------
 class TaskAlignedAssigner(nn.Module):
+    """
+        This code referenced to https://github.com/ultralytics/ultralytics
+    """
     def __init__(self,
                  num_classes     = 80,
                  topk_candidates = 10,

yolo/models/yolov8_e2e/matcher.py

@@ -6,6 +6,9 @@ from utils.box_ops import bbox_iou
 
 # -------------------------- Task Aligned Assigner --------------------------
 class TaskAlignedAssigner(nn.Module):
+    """
+        This code referenced to https://github.com/ultralytics/ultralytics
+    """
     def __init__(self,
                  num_classes     = 80,
                  topk_candidates = 10,