昇腾社区首页
中文
注册

(beta)torch_npu.contrib.module.Focus

函数原型

[object Object]

功能说明

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

参数说明

  • c1 (Int) - 输入图像中的通道数。
  • c2 (Int) - 卷积产生的通道数。
  • k(Int,默认值为1) - 卷积核大小。
  • s(Int,默认值为1) - 卷积步长。
  • p (Int) - 填充。
  • g (Int,默认值为1) - 从输入通道到输出通道的分组数。
  • act (Bool,默认值为True) - 是否使用激活函数。

支持的型号

  • [object Object]Atlas 训练系列产品[object Object]
  • [object Object]Atlas A2 训练系列产品[object Object]
  • [object Object]Atlas A3 训练系列产品[object Object]
  • [object Object]Atlas 推理系列产品[object Object]

调用示例

[object Object]