yolov5_obb
时间: 2023-08-06 15:05:40 浏览: 50
yolov5_obb是一款用于斜框目标检测的神经网络。它是基于yolov5框架改写而来的,主要用于检测斜框形状的目标。具体原理可以在相关的中文博客中找到详细的介绍。作者也在知乎上发布了一系列关于如何改写ultralytics大佬的yolov5用于斜框检测的文章。\[1\] 你可以按照这些教程一步一步地进行标图、部署和使用。在进行部署时,你需要使用tensorrt来生成yolov5_obb的engine文件。在执行部署命令时,你需要指定相应的权重文件和输入图片。\[2\]\[3\]
#### 引用[.reference_title]
- *1* *2* *3* [Yolov5旋转框(斜框)检测tensorrt部署(C++)从入门到入坟](https://blog.csdn.net/qq_41043389/article/details/127777272)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]
相关问题
YOLOv5_obb
YOLOv5_obb是一款用于斜框目标检测的神经网络,它是基于YOLOv5框架改写而来的。该网络的具体原理可以在相关的中文博客中找到,也有作者在知乎上发布了一系列关于如何改写ultralytics大佬的YOLOv5用于斜框检测的文章。\[2\]要部署YOLOv5_obb,首先需要在终端运行"make"命令,这将在build文件夹中生成一个名为"yolov5_gen"的可执行文件(在Windows中为exe文件)。然后,在终端中运行以下命令来执行YOLOv5_obb的推理过程:sudo ./yolov5_gen -s ../yolov5s.wts ../yolov5s.engine s。\[1\]这样就可以使用YOLOv5_obb对斜框目标进行检测了。
#### 引用[.reference_title]
- *1* *2* *3* [Yolov5旋转框(斜框)检测tensorrt部署(C++)从入门到入坟](https://blog.csdn.net/qq_41043389/article/details/127777272)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]
yolov5_obb安装
安装yolov5_obb需要进行以下几个步骤:
1. 安装CUDA。这是深度学习中必要的计算框架。如果你已经安装了CUDA,可以跳过此步骤。如果没有安装,需要去官网下载并安装。注意,需要安装与你的显卡对应的CUDA版本。
2. 下载yolov5_obb代码。可以从github上下载对应的代码包,下载链接为 https://github.com/hukaixuan19970627/yolov5_obb。
3. 安装依赖项。在yolov5_obb-master目录下打开命令行窗口,执行以下命令:pip install -r requirements.txt
4. 安装旋转框非极大值抑制算法。在yolov5_obb-master/utils/nms_rotated目录下执行以下命令:python setup.py develop
安装完成后即可开始使用yolov5_obb进行物体检测。需要注意的是,如果在安装和使用过程中遇到了问题,可以参考引用中提供的资源进行解决。
--相关问题--: