YOLOV8 改进——添加CA注意力机制

参考博客：YOLOv8改进算法之添加CA注意力机制_yolov8添加注意力机制-CSDN博客

在ultralytics/cfg/datasets/ 下新建data.yaml

path: D:/Desktop/fire_smoke_dataset(1)
train: D:/Desktop/fire_smoke_dataset(1)/images/train

val: D:/Desktop/fire_smoke_dataset(1)/images/val

nc : 2
names:
  0: fire
  1: smoke

main.py训练：

#coding:utf-8
from ultralytics import YOLO


model = YOLO("ultralytics/cfg/models/v8/yolov8s-CA.yaml").load('yolov8n.pt')

# Use the model
if __name__ == '__main__':
    # Use the model
    results = model.train(data='ultralytics/cfg/datasets/data.yaml', epochs=100, batch=4,device='0',workers=1,amp=False)  # 训练模型
    # 将模型转为onnx格式
    # success = model.export(format='onnx')

训练结果：

该训练结果与没有改进的结果差不多，在我的数据集上没有改进。下图是没有改进的训练结果。