# YOLOv5m_SE_CBAM
Based on YOLOv5-v6.0 version, using YOLOv5m model to add attention mechanism (SE, CBAM).
I deleted the model I have trained in the zip file, if you want to use the pre-trained model, please go to https://github.com/ultralytics/yolov5 to download the pt file, you can also train yourself, SE module and I have added the CBAM module.