主要就是制作数据集,数据集有几百张阀门图片,通过数据增强技术,比如旋转、去噪、缩放、裁剪等等把数据集扩到了1万张左右。在yolo8中引入了注意力机制和EfficientNet2,注意力机制有CA、CBAM、GAM、SE以及这些注意力机制和EfficientNet2的结合,训练后生成模型,然后使用模型进行预测!
主要就是制作数据集,数据集有几百张阀门图片,通过数据增强技术,比如旋转、去噪、缩放、裁剪等等把数据集扩到了1万张左右。在yolo8中引入了注意力机制和EfficientNet2,注意力机制有CA、CBAM、GAM、SE以及这些注意力机制和EfficientNet2的结合,训练后生成模型,然后使用模型进行预测!