yolov5加入注意力机制步骤

yolov5加入注意力机制步骤

  1. common.py
  2. 、yolo.py
  3. .创建自定义的yaml文件
  4. 验证加入注意力机制是否正确

总结
yolov5加入注意力机制步骤
1.common.py添加相应条件
2.yolo.py添加判断条件
3.创建属于自己的注意力yaml文件

提示:以下是本篇文章正文内容,下面案例可供参考

一、common.py
在common.py中先添加你想添加的注意力模块

class h_sigmoid(nn.Module):
    def __init__(self, inplace=True):
        super

你可能感兴趣的