Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

pading from (k-1) // 2 to k // 2

This commit is contained in:
Glenn Jocher 2020-04-11 12:18:54 -07:00
parent b574f765ce
commit a34219a54b
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -28,7 +28,7 @@ def create_modules(module_defs, img_size):
out_channels=filters, out_channels=filters,
kernel_size=size, kernel_size=size,
stride=stride, stride=stride,
padding=(size - 1) // 2 if mdef['pad'] else 0, padding=size // 2 if mdef['pad'] else 0,
groups=mdef['groups'] if 'groups' in mdef else 1, groups=mdef['groups'] if 'groups' in mdef else 1,
bias=not bn)) bias=not bn))
else: # multiple-size conv else: # multiple-size conv

2
utils/layers.py
View File

@ -65,7 +65,7 @@ class MixConv2d(nn.Module): # MixConv: Mixed Depthwise Convolutional Kernels ht
out_channels=ch[g], out_channels=ch[g],
kernel_size=k[g], kernel_size=k[g],
stride=stride, stride=stride,
padding=(k[g] - 1) // 2, # 'same' pad padding=k[g] // 2, # 'same' pad
dilation=dilation, dilation=dilation,
bias=bias) for g in range(groups)]) bias=bias) for g in range(groups)])