添加一下模块

没用，看不懂

工具：Netron

需要用export.py对pt文件进行导出

Usage: $ python export.py --weights yolov5s.pt --include torchscript onnx openvino engine coreml tflite ...

这里导出为onnx格式，直接适配Netron工具

pip install onnx
python export.py --weight yolov5s.pt --include onnx

添加注意力机制

具体怎么修改可以直接用double-vin/yolov5_attention (github.com)cang’l仓库

SENet

和使用yolov5官方的网络构架对比来说，加入se注意力机制模块后，就目前状况下，出现了一定的问题。

背景：15张刨花板图片，不添加任何模块的情况下，1.4hour+完成了训练。结果见yoloV5_results

在SPPF前加入se模块后，在第222轮结束后出现了：Stopping training early as no improvement observed in last 100 epochs. Best results observed at epoch 122, best model saved as best.pt.

而且原本是通过增加少量的计算成本提升精度，就我的情况来说是降低了计算成本，也降低了精度。

结果也是非常的糟糕。

C3SE

1.612hours

而没有添加任何模块的结果或许收敛的更好

我觉得，se模块说通过对全局信息的学习，去加强对象的特征，可能我全局全是刨花，才效果不理想的。

更新

需要删除yolov5自带的Mosaic数据增强才有效果！