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