class Focus()

class Focus(nn.Module):

使用NPU亲和写法替换YOLOv5中的原生Focus。

• 参数：
  • c1 (Int) - 输入图像中的通道数。
  • c2 (Int) - 卷积产生的通道数。
  • k(Int，默认值为1) - 卷积核大小。
  • s(Int，默认值为1) - 卷积步长。
  • p (Int) - 填充。
  • g (Int，默认值为1) - 从输入通道到输出通道的阻塞连接数。
  • act (Bool，默认值为True) - 是否使用激活函数。
• 调用方式样例：

  from torch_npu.contrib.module import Focus
  fast_focus = Focus(8, 13).npu()

• 示例：

     >>> input = torch.randn(4, 8, 300, 40).npu()
     >>> input.requires_grad_(True)
     >>> fast_focus = Focus(8, 13).npu()
     >>> output = fast_focus(input)
     >>> output.sum().backward()