复现 Oriented R-CNN RTX3080ti

最新推荐文章于 2023-05-29 13:21:15 发布

pinfan12138

最新推荐文章于 2023-05-29 13:21:15 发布

阅读量3k

点赞数 6

文章标签： python 目标检测 linux 深度学习

本文链接：https://blog.youkuaiyun.com/pinfan12138/article/details/125805836

版权

本文详细记录了在RTX 3080ti显卡上、Ubuntu 20.04系统中复现Oriented R-CNN的过程。通过创建合适环境，解决torch与CUDA版本兼容问题，安装OBBdetection、BboxToolkit和mmcv，最终成功运行代码并展示了训练与验证的效果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在3080ti主机上复现Oriented R-CNN，参考了：复现Oriented R-CNN RTX 2080Ti_吃肉不能购的博客-优快云博客

这篇博文给我提供了很大帮助

前言

GitHub链接：GitHub - jbwang1997/OBBDetection: OBBDetection is an oriented object detection library, which is based on MMdetection.

显卡：RTX3080ti

系统：Ubuntu20.04

一、创建环境

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

这里我一开始用了我的比较高版本的pytorch去跑，发现会报THC/THC.h的错误，简单说就是torch版本太高，和代码里用的函数不匹配。之后参考另一篇博文想用pytorch=1.6.0，但是30系显卡只支持cuda11，所以尝试了最低的支持cuda11的版本torch=1.7.0，然后代码setup成功了，但还是莫名其妙有错误，查了说好像需要1.7.1。第二天索性删掉环境重来，直接下载torch1.7.1。

二、配置OBBdetection

下载代码文件

git clone https://github.com/jbwang1997/OBBDetection.git --recursive
cd OBBDetection

安装BboxToolkit

cd BboxToolkit
pip install -v -e .  
cd ..

如果下载git下载的时候出错误了BboxToolkit没下载下来，或者文件夹没有setup.py文件，就去https://github.com/jbwan

最低0.47元/天解锁文章