diff --git a/models/yolo.py b/models/yolo.py index 9a5b8da..ae50c85 100644 --- a/models/yolo.py +++ b/models/yolo.py @@ -149,7 +149,7 @@ def parse_model(md, ch): # model_dict, input_channels(3) pass n = max(round(n * gd), 1) if n > 1 else n # depth gain - if m in [nn.Conv2d, Conv, Bottleneck, SPP, DWConv, MixConv2d, Focus, ConvPlus, BottleneckCSP, BottleneckLight]: + if m in [nn.Conv2d, Conv, Bottleneck, SPP, DWConv, MixConv2d, Focus, ConvPlus, BottleneckCSP]: c1, c2 = ch[f], args[0] # Normal