yapf reformat

models/common.py

@@ -365,8 +365,9 @@ def __init__(self, weights='yolov5s.pt', device=torch.device('cpu'), dnn=False,
             model.half() if fp16 else model.float()
             if extra_files['config.txt']:  # load metadata dict
                 d = json.loads(extra_files['config.txt'],
-                               object_hook=lambda d: {int(k) if k.isdigit() else k: v
-                                                      for k, v in d.items()})
+                               object_hook=lambda d: {
+                                   int(k) if k.isdigit() else k: v
+                                   for k, v in d.items()})
                 stride, names = int(d['stride']), d['names']
         elif dnn:  # ONNX OpenCV DNN
             LOGGER.info(f'Loading {w} for ONNX OpenCV DNN inference...')
@@ -884,6 +885,7 @@ def forward(self, x):
 
 # contributed by @aash1999
 class ChannelAttention(nn.Module):
+
     def __init__(self, in_planes, ratio=16):
         """
         Initialize the Channel Attention module.
@@ -920,6 +922,7 @@ def forward(self, x):
 
 # contributed by @aash1999
 class SpatialAttention(nn.Module):
+
     def __init__(self, kernel_size=7):
         """
         Initialize the Spatial Attention module.
@@ -996,6 +999,7 @@ def forward(self, x):
 
 # contributed by @aash1999
 class Involution(nn.Module):
+
     def __init__(self, c1, c2, kernel_size, stride):
         """
         Initialize the Involution module.