|
|
@@ -19,8 +19,8 @@ from utils.torch_utils import time_synchronized
|
|
|
|
|
|
#错误改正
|
|
|
#新增类
|
|
|
-import warnings
|
|
|
|
|
|
+import warnings
|
|
|
class SPPF(nn.Module):
|
|
|
# Spatial Pyramid Pooling - Fast (SPPF) layer for YOLOv5 by Glenn Jocher
|
|
|
def __init__(self, c1, c2, k=5): # equivalent to SPP(k=(5, 9, 13))
|
|
|
@@ -40,7 +40,6 @@ class SPPF(nn.Module):
|
|
|
|
|
|
|
|
|
|
|
|
-
|
|
|
def autopad(k, p=None): # kernel, padding
|
|
|
# Pad to 'same'
|
|
|
if p is None:
|