news 2026/5/8 21:10:45

OBBDetection环境配置步骤

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
OBBDetection环境配置步骤

本文仅涉及OBBDetection的环境配置,因为我要复现Oriented R-CNN for Object Detection,具体怎样训练自己的数据集网上有好多攻略…
我对OBBDetection环境进行配置的时候出现了很多错误,导致一直配不成功,看了别的大佬的解决办法也没用,用这种方式配好了,给大家一种参考吧。

个人是租的autodl显卡RTX4090,网上很多步骤都是3080ti,我使用4090多加了一行步骤也可以。

一、创建环境

conda create -n orcnn python=3.7 -y conda activate orcnn conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

cuda使用的11.0,因为我发现太新的几乎必失败,可以按照我这个环境版本来。

二、配置OBBDetection的BboxToolkit

下载后的OBBDetection其实有两部分,一个是OBBDetection,一个是BboxToolkit,这里下载好代码后,要进入OBBDetection/BboxToolkit目录下安装BbokToolkit的依赖,否则后面会报错。

下载代码: git clone https://github.com/jbwang1997/OBBDetection.git --recursive cd OBBDetection cd BboxToolkit pip install -v -e . cd ..

安装完BboxToolkit的依赖后回到OBBDetection的文件夹路径下

三、安装mmcv

很多都是这里报错,除了版本对应,还有可能需要调整显卡的识别版本。
这里官方其实推荐mmcv-full要1.3以上,最新的默认1.6,实测会报错,用1.4.0可以。

pip install mmcv-full==1.4.0 -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html --no-cache-dir
  1. 后面官方推荐安的是mmpycocotools,但是好像已经废弃了,用pycocotools来替代。
  2. 运行export TORCH_CUDA_ARCH_LIST="8.0"来调整显卡识别版本,30系显卡好像可以不用运行。
pip install -r requirements/build.txt pip install pycocotools export TORCH_CUDA_ARCH_LIST="8.0" pip install -v -e .

最后运行pip install -v -e .会有点慢,而且输出很多格式很混乱的文字,不用管,只要最后出现Successfully installed Pillow6.2.2 mmdet-2.2.0+cb0cb7a就是成功了。

四、下载预训练权重用于测试

下载他给的预训练权重,地址为:https://github.com/jbwang1997/OBBDetection/tree/master/configs/obb/oriented_rcnn


OBBDetection文件夹下创建一个新的文件夹ckpt,用来保存预训练权重文件,把刚才下载的预训练文件(.pth后缀)放到这里面。
输入运行代码:

python demo/huge_image_demo.py demo/dota_demo.jpg configs/obb/oriented_rcnn/faster_rcnn_orpn_r101_fpn_1x_ms_rr_dota10.py ckpt/faster_rcnn_orpn_r101_fpn_1x_dota10_epoch12.pth BboxToolkit/tools/split_configs/dota1_0/ss_test.json

注意这里代码有一个:kpt/faster_rcnn_orpn_r101_fpn_1x_dota10_epoch12.pth,这里的faster_rcnn_orpn_r101_fpn_1x_dota10_epoch12.pth换成你下载的预训练权重文件名,我下载的第一个就是这个,就不用改了。

五、训练demo测试

运行上面的代码后会出现如下:

这就是环境配置成功了,下面的步骤让它显示demo图像,更清晰直观一点。
找到OBBDetection中的mmdet/apis/inference.py,修改代码:

# 在开头导入importcv2# 找到最后一个函数:show_result_pyplot,在这个函数里的最后一行加入:cv2.imwrite("xxxx320.jpg",img)

重新运行代码:

python demo/huge_image_demo.py demo/dota_demo.jpg configs/obb/oriented_rcnn/faster_rcnn_orpn_r101_fpn_1x_ms_rr_dota10.py ckpt/faster_rcnn_orpn_r101_fpn_1x_dota10_epoch12.pth BboxToolkit/tools/split_configs/dota1_0/ss_test.json

最后会在OBBDetection的根目录生成图片xxxx320.jpg,打开如下就是配成功了。

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/5/8 21:04:20

Stackmoss:构建生产级AI原生应用的一体化框架实战指南

1. 项目概述与核心价值最近在开源社区里,Stackmoss 这个项目引起了我的注意。它不是一个简单的工具库,而是一个旨在构建“AI原生应用”的完整技术栈。简单来说,它想解决的问题是:当你想开发一个真正由AI驱动、而非仅仅调用API的应…

作者头像 李华
网站建设 2026/5/8 21:03:35

从无人机到3D打印机:2804电机参数测量背后的FOC控制实战意义

从无人机到3D打印机:2804电机参数测量背后的FOC控制实战意义 在穿越机竞速比赛中,0.1秒的响应延迟可能导致撞杆;在高精度3D打印中,5%的扭矩波动会留下可见的层纹;当小型机器人关节驱动出现10的位置误差时,抓…

作者头像 李华
网站建设 2026/5/8 21:02:29

IE项目迁移Chrome避坑指南:不只是ActiveXObject,这些API也要小心

IE项目迁移Chrome避坑指南:不只是ActiveXObject,这些API也要小心 当企业决定将老旧Web系统从IE迁移到Chrome时,技术团队往往会发现ActiveXObject报错只是冰山一角。实际上,IE特有的API和行为差异遍布整个代码库,从事件…

作者头像 李华
网站建设 2026/5/8 20:59:44

别只吐槽Broadcom!VMware迁移后,这些隐藏的‘宝藏’网站和工具依然能用

VMware迁移后依然可用的隐藏资源与实用工具指南 当Broadcom完成对VMware的收购后,许多用户发现熟悉的界面和访问方式发生了改变,这确实带来了一些不便。但值得庆幸的是,VMware生态系统中那些真正有价值的核心资源——那些我们日常工作中依赖…

作者头像 李华