modify yolov6

yolo/models/yolov6/yolov6_backbone.py

@@ -2,9 +2,9 @@ import torch
 import torch.nn as nn
 
 try:
-    from .yolov6_basic import RepBlock, RepVGGBlock, RepCSPBlock
+    from .modules import RepBlock, RepVGGBlock, RepCSPBlock
 except:
-    from  yolov6_basic import RepBlock, RepVGGBlock, RepCSPBlock
+    from  modules import RepBlock, RepVGGBlock, RepCSPBlock
 
 # IN1K pretrained weight
 pretrained_urls = {

yolo/models/yolov6/yolov6_head.py

@@ -2,9 +2,9 @@ import torch
 import torch.nn as nn
 
 try:
-    from .yolov6_basic import BasicConv
+    from .modules import BasicConv
 except:
-    from  yolov6_basic import BasicConv
+    from  modules import BasicConv
 
 
 ## Single-level Detection Head

yolo/models/yolov6/yolov6_neck.py

@@ -2,9 +2,9 @@ import torch
 import torch.nn as nn
 
 try:
-    from .yolov6_basic import BasicConv
+    from .modules import BasicConv
 except:
-    from  yolov6_basic import BasicConv
+    from  modules import BasicConv
     
 
 # Spatial Pyramid Pooling - Fast (SPPF) layer for YOLOv5 by Glenn Jocher

yolo/models/yolov6/yolov6_pafpn.py

@@ -4,9 +4,9 @@ import torch.nn as nn
 import torch.nn.functional as F
 
 try:
-    from .yolov6_basic import BasicConv, RepBlock, RepCSPBlock
+    from .modules import BasicConv, RepBlock, RepCSPBlock
 except:
-    from  yolov6_basic import BasicConv, RepBlock, RepCSPBlock
+    from  modules import BasicConv, RepBlock, RepCSPBlock
 
 
 # Yolov6FPN