Yi_Zhang
GitHub
4 years ago
1 changed files with 2 additions and 2 deletions
+ 2
- 2
models/experimental.py
View File
| @@ -67,8 +67,8 @@ class GhostConv(nn.Module): | |||
| def __init__(self, c1, c2, k=1, s=1, g=1, act=True): # ch_in, ch_out, kernel, stride, groups | |||
| super(GhostConv, self).__init__() | |||
| c_ = c2 // 2 # hidden channels | |||
| self.cv1 = Conv(c1, c_, k, s, g, act) | |||
| self.cv2 = Conv(c_, c_, 5, 1, c_, act) | |||
| self.cv1 = Conv(c1, c_, k, s, None, g, act) | |||
| self.cv2 = Conv(c_, c_, 5, 1, None, c_, act) | |||
| def forward(self, x): | |||
| y = self.cv1(x) | |||
Loading…