记录使用yolov5进行旋转目标的检测

由于实习公司需要使用到旋转目标的检测,所以这几天学习了相关知识,并找了许多资料,饶了许多的弯路。下面记录下项目的整个实现过程。

我参考的是以下几位博主:

DOTAv2遥感图像旋转目标检测竞赛经验分享(Swin Transformer + Anchor free/based方案) - 知乎

小鸡炖技术的个人空间_哔哩哔哩_Bilibili

首先,先为该项目创建一个虚拟环境

VS2015

pytorch 1.6.0(其他版本也可以进行尝试)

torchvision 0.7.0(其他版本也可以进行尝试)

其中这个VS2015,17,19版本都可以,但是pytorch和torchvision必须是要求的版本 ,不然可能会出错 这里试了torch1.8.1和torchvision0.9.1是可以用的。建议安装pytorch和torchvision时采用离线安装包安装,比较稳定。地址,

https://download.pytorch.org/whl/torch_stable.html

接着打开Anaconda prompt创建项目的虚拟环境

conda create -n royolov5 python=3.8
conda activate yolov5

进入创建的虚拟环境之后cd进入下载好pytorch和torchvision的文件夹。然后执行

pip install torch-1.6.0+cu101-cp38-cp38-win_amd64.whl

pip install torchvision-0.7.0+cu101-cp38-cp38-win_amd64.whl

这里注意后缀名.whl不要忘了。

到这里,就将pytorch框架搭好了。

接着,下载我发的压缩包,解压后,同样cd进入YOLOv5_DOTA_OBB-master项目文件的主目录

执行 pip install -r requirements.txt 安装相关的包。

再,添加E:\PytorchPro\YOLOv5_DOTA_OBB-master\swigwin-4.0.2到path里面,这里要进行系统环境的刷新,可以选择重启,也可以进入cmd,输入set path=c 进行刷新

再,在Anaconda prompt中的royolov5环境下cd进入utils分别运行

swig -c++ -python polyiou.i

python setup.py build_ext --inplace

再,将E:\ProgramData\Anaconda3\envs\royolov5\Lib\site-packages\shapely\DLLs所有dll复制到E:\ProgramData\Anaconda3\envs\royolov5\Library\bin

到这一步,训练环境基本准备好了,下面开始准备旋转目标所需的数据集

首先要使用到rolabelimg旋转目标标注工具,具体怎么使用自行百度。下载地址

https://github.com/cgvict/roLabelImg

标注图像后得到的是xml格式,这里需要对数据格式进行转换,符合yolov5的txt格式

首先,运行roxml_to_data.py,转换到Dota的8点数据格式

再将图像数据依次放在以下文件夹

4d8fc48a684c42a486f50501aa23a6d2.png

其中images里存放原图像,labelTxt中存放上一步转化的8点格式txt文件(注意要和images中的图像文件对应) 

再,进入DOTA_devkit_YOLO-master,修改DOTA_devkit_YOLO-master\dota_utils.py中的classnames_v1_5,修改为你标注数据的种类

87f5aa03ee6b44b1b87d88980df7c5f5.png

再,分别运行PaddingPIC.py和YOLO_Transform.py

其中,PaddingPIC.py的作用是将图片扩展到高宽相同,便于数据处理。YOLO_Transform.py的作用是将dota的数据格式转换为yolov5的数据格式。

转换后的txt文件会存放再yolo_labels文件夹中

再,将yolov5训练所需的原图像和上一步操作得到的txt文件分别放入DOTA_demo_view文件夹里的images和labels中

78a298a3e6f14024b8b024d76063a572.png

 

再修改data\DOTA_ROTATED.yaml中nc和names,或者自己写个yaml文件

可以再修改下网络模型的yaml文件,不会的话用官方的yaml文件就行

最后别忘了下载权重文件,这里要注意,因为这个项目是基于v5-3.1版本的,所以,下载的权重文件不能下载最新的pt文件,不然会报错。

最后,开始训练就可以了。

 

给出资源链接:

【超级会员V3】通过百度网盘分享的文件:JLJ.zip
链接:https://pan.baidu.com/s/17zVi6ae6sSs3ab7cUiWhig 
提取码:8e3g
 

 

 

 

 

 

 


http://www.niftyadmin.cn/n/5097107.html

相关文章

iOS——JSONModel的使用与JSONModel的嵌套

什么是JSONModel JSONModel是一个解析JSON数据的开源库,可以将JSON数据直接解析成自定义的model 使用 JSONModel 非常简单,只需要将你的 model 类继承自 JSONModel ,而同时 model 中的属性名又恰巧可以和 JSON 数据中的 key 名字一样的话,那么非常恭喜你,你的工作…

如何将前后端分离项目部署到本地的Docker Desktop容器运行并且访问

文章目录 前言 完成了客户的一个前后端分离项目,要求部署到客户电脑上去展示,那肯定不能直接把代码弄上去跑呀~~~,于是我就想把他们都打包部署到本地的docker容器里面,方便运行和访问,so,以下内容就详细介…

Adaptive AUTOSAR CM模块介绍(六)

Event订阅和缓存原理 Event事件并不是proxy定义了就可以使用,需要Event事件的订阅,一旦订阅成功了,有event事件的时候,就会收到订阅的消息。 /* brief The application expects the CM to subscribe the event. param maxSampleC…

XTU-OJ 1412-Rotate Again

大家可能也发现我是实时更新的,所以大家可以点赞、评论,如果讲解的不够详细或者看不懂的都可以留言。多给我些反馈,可以让我更好的编写博客内容。(但是别催更啊~~) 题目描述 一个44的矩阵,我们可以把矩阵…

SQL实现自定义排序

业务事实上,经常会遇到排序问题,对数值类型字段排序不会有很大争议,但是有时希望对字符型字段按要求进行排序。本文记录自定义排序的几种方法。 使用case when关键字进行限定 本方法思路是将希望排序小的字符取值为较小的数字。注意end后面…

0基础python全栈教程前言,0基础学习python难不难

近年来,随着人工智能的发展和高新科技的进步,python的火热程度有目共睹,其简洁的语法和丰富的第三方库,以及人工智能的巨大潜力和巨大的市场,让其未来充满了想象空间。 0基础学习python难不难 这个并不好说&#xff…

Windows系统下设置程序开机自启(WinSW)

介绍 WinSW可以将Windows上的任何程序作为系统服务进行管理,已达到开机自启的效果。 支持的平台 WinSW需要运行在拥有.NET Framework 4.6.1或者更新版本的Windows平台下 下载 github: 下载地址 百度网盘(v2.12.0)&#xff1a…

最短路问题

本文用到了Dijkstra算法和Floyd算法,要想了解这两种算法可看Floyd算法和Dijkstra算法 目录 一、最短路问题简述 二、求定点间的最短距离(Dijkstra算法) 1、Dijkstra(笛卡斯特拉)算法(1959年)原理 2、符…