update

models/yolov1/yolov1_neck.py

@@ -5,6 +5,9 @@ from .yolov1_basic import Conv
 
 # Spatial Pyramid Pooling - Fast (SPPF) layer for YOLOv5 by Glenn Jocher
 class SPPF(nn.Module):
+    """
+        This code referenced to https://github.com/ultralytics/yolov5
+    """
     def __init__(self, in_dim, out_dim, expand_ratio=0.5, pooling_size=5, act_type='lrelu', norm_type='BN'):
         super().__init__()
         inter_dim = int(in_dim * expand_ratio)

models/yolov2/yolov2_neck.py

@@ -5,6 +5,9 @@ from .yolov2_basic import Conv
 
 # Spatial Pyramid Pooling - Fast (SPPF) layer for YOLOv5 by Glenn Jocher
 class SPPF(nn.Module):
+    """
+        This code referenced to https://github.com/ultralytics/yolov5
+    """
     def __init__(self, in_dim, out_dim, expand_ratio=0.5, pooling_size=5, act_type='lrelu', norm_type='BN'):
         super().__init__()
         inter_dim = int(in_dim * expand_ratio)

models/yolov3/yolov3_neck.py

@@ -5,6 +5,9 @@ from .yolov3_basic import Conv
 
 # Spatial Pyramid Pooling - Fast (SPPF) layer for YOLOv5 by Glenn Jocher
 class SPPF(nn.Module):
+    """
+        This code referenced to https://github.com/ultralytics/yolov5
+    """
     def __init__(self, in_dim, out_dim, expand_ratio=0.5, pooling_size=5, act_type='lrelu', norm_type='BN'):
         super().__init__()
         inter_dim = int(in_dim * expand_ratio)

models/yolov4/yolov4_neck.py

@@ -5,6 +5,9 @@ from .yolov4_basic import Conv
 
 # Spatial Pyramid Pooling - Fast (SPPF) layer for YOLOv5 by Glenn Jocher
 class SPPF(nn.Module):
+    """
+        This code referenced to https://github.com/ultralytics/yolov5
+    """
     def __init__(self, in_dim, out_dim, expand_ratio=0.5, pooling_size=5, act_type='lrelu', norm_type='BN'):
         super().__init__()
         inter_dim = int(in_dim * expand_ratio)

models/yolov5/yolov5_neck.py

@@ -5,6 +5,9 @@ from .yolov5_basic import Conv
 
 # Spatial Pyramid Pooling - Fast (SPPF) layer for YOLOv5 by Glenn Jocher
 class SPPF(nn.Module):
+    """
+        This code referenced to https://github.com/ultralytics/yolov5
+    """
     def __init__(self, in_dim, out_dim, expand_ratio=0.5, pooling_size=5, act_type='lrelu', norm_type='BN'):
         super().__init__()
         inter_dim = int(in_dim * expand_ratio)

models/yolov7/matcher.py

@@ -1,11 +1,18 @@
+# ---------------------------------------------------------------------
+# Copyright (c) Megvii Inc. All rights reserved.
+# ---------------------------------------------------------------------
+
+
 import torch
 import torch.nn.functional as F
 from utils.box_ops import *
 
 
-
 # YOLOX SimOTA
 class SimOTA(object):
+    """
+        This code referenced to https://github.com/Megvii-BaseDetection/YOLOX/blob/main/yolox/models/yolo_head.py
+    """
     def __init__(self, 
                  num_classes,
                  center_sampling_radius,

models/yolov7/yolov7_neck.py

@@ -5,6 +5,9 @@ from .yolov7_basic import Conv
 
 # Spatial Pyramid Pooling - Fast (SPPF) layer for YOLOv5 by Glenn Jocher
 class SPPF(nn.Module):
+    """
+        This code referenced to https://github.com/ultralytics/yolov5
+    """
     def __init__(self, in_dim, out_dim, expand_ratio=0.5, pooling_size=5, act_type='lrelu', norm_type='BN'):
         super().__init__()
         inter_dim = int(in_dim * expand_ratio)

models/yolov8/yolov8_neck.py

@@ -8,6 +8,9 @@ except:
 
 # Spatial Pyramid Pooling - Fast (SPPF) layer for YOLOv5 by Glenn Jocher
 class SPPF(nn.Module):
+    """
+        This code referenced to https://github.com/ultralytics/yolov5
+    """
     def __init__(self, in_dim, out_dim, expand_ratio=0.5, pooling_size=5, act_type='', norm_type=''):
         super().__init__()
         inter_dim = int(in_dim * expand_ratio)

models/yolox/matcher.py

@@ -1,11 +1,18 @@
+# ---------------------------------------------------------------------
+# Copyright (c) Megvii Inc. All rights reserved.
+# ---------------------------------------------------------------------
+
+
 import torch
 import torch.nn.functional as F
 from utils.box_ops import *
 
 
-
 # YOLOX SimOTA
 class SimOTA(object):
+    """
+        This code referenced to https://github.com/Megvii-BaseDetection/YOLOX/blob/main/yolox/models/yolo_head.py
+    """
     def __init__(self, 
                  num_classes,
                  center_sampling_radius,

models/yolox/yolox_neck.py

@@ -9,6 +9,9 @@ except:
 
 # Spatial Pyramid Pooling - Fast (SPPF) layer for YOLOv5 by Glenn Jocher
 class SPPF(nn.Module):
+    """
+        This code referenced to https://github.com/ultralytics/yolov5
+    """
     def __init__(self, in_dim, out_dim, expand_ratio=0.5, pooling_size=5, act_type='', norm_type=''):
         super().__init__()
         inter_dim = int(in_dim * expand_ratio)