我的YOLOv5版本為6.0

需要修改的兩個文件如下：

• ./models/yolo5s_SE-Net.yaml
• ./models/commom.py

1. yolo5s_SE-Net.yaml文件

yolo5s_SE-Net.yaml 修改自文件yolo5s.yaml

yolo5s.yaml 文件內容為：

# YOLOv5 ?? by Ultralytics, GPL-3.0 license# Parameters nc: 1 # number of classes depth_multiple: 0.33 # model depth multiple width_multiple: 0.50 # layer channel multiple anchors:

總結

以上是生活随笔為你收集整理的YOLOv5添加注意力机制 Pytorch的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。