Pārlūkot izejas kodu

module updates

5.0
Glenn Jocher pirms 4 gadiem
vecāks
revīzija
715cb08b10
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  1. +1
    -1
      models/yolo.py

+ 1
- 1
models/yolo.py Parādīt failu

@@ -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

Notiek ielāde…
Atcelt
Saglabāt