class Focus()

API接口

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()

父主题： torch_npu.contrib