(self, in_channels, out_channels, ksize=1, stride=1, act="silu")
| 189 | """Focus width and height information into channel space.""" |
| 190 | |
| 191 | def __init__(self, in_channels, out_channels, ksize=1, stride=1, act="silu"): |
| 192 | super().__init__() |
| 193 | self.conv = BaseConv(in_channels * 4, out_channels, ksize, stride, act=act) |
| 194 | |
| 195 | def forward(self, x): |
| 196 | # shape of x (b,c,w,h) -> y(b,4c,w/2,h/2) |