zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

module updates

This commit is contained in:
Glenn Jocher 2020-06-08 22:15:16 -07:00
parent ff02ae0869
commit 715cb08b10
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models/yolo.py
View File

@ -149,7 +149,7 @@ def parse_model(md, ch): # model_dict, input_channels(3)
pass pass
n = max(round(n * gd), 1) if n > 1 else n # depth gain 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] c1, c2 = ch[f], args[0]
# Normal # Normal