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,所以还没有实现最理想的效果。

训练损失:

在这里插入图片描述
在这里插入图片描述

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

猜你喜欢

转载自blog.csdn.net/weixin_41981679/article/details/125961255