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格式
yolov5实现旋转目标检测_第1张图片
yolov5实现旋转目标检测_第2张图片

检测效果:



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

训练损失:

yolov5实现旋转目标检测_第3张图片
yolov5实现旋转目标检测_第4张图片

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

你可能感兴趣的:(目标检测,目标检测,深度学习,python)