yolov5实现旋转目标检测

基于yolov5实现旋转目标检测：

`提示：通过添加角度回归实现旋转的目标检测， 对于角度部分的回归使用kld loss实现

---

学习时间：

提示：[此项目是基于知乎作者胡凯旋的自己改建YOLOv5旋转目标的踩坑记录

](https://zhuanlan.zhihu.com/p/358072483)

---

安装环境：

pip install -r requirements.txt
cd utils/nms_rotated
python setup.py develop #or “pip install -v -e .”

cd yolov5_obb/DOTA_devkit
sudo apt-get install swig
swig -c++ -python polyiou.i
python setup.py build_ext --inplace

准备数据集：

数据集的标签是基于dota格式，所以在自定义数据集的时候需要先转换数据集的格式为dota格式

检测效果：

此项目只训练100 epoch，所以还没有实现最理想的效果。

训练损失：

如果有需要源码的小伙伴可以私信进行获取