最近在看一些基于点云3d目标检测的文章,需要复现甚至修改一些算法,就找到了mmlab开源的mmdetection3d目标检测框架,方便后续学习。
在安装的时候遇到一点坑,比如环境问题,安装完能跑demo但是不能跑训练测试问题等。在解决问题后还是完成了安装。在这里记录一下正确的安装流程,已备再次查阅,也给大家提供一点参考。
首先建环境,最好是新建一个环境
conda create -n det python=3.8
#激活
conda activate det
注意,以下步骤均在环境下进行!
然后就是装pytorch,一定要根据cuda的版本来选择。用nvidia-smi查看可cuda版本, torch的网址:pytorch; 什么?还有没装cuda的?cuda的网站:cuda;
我的cuda版本是10.2,所以选择torch的安装命令的时候注意:
conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=10.2 -c pytorch
然后就正式进入mmdetection3d的依赖的安装环节了,首先下载工具mim:
pip install openmim
然后一次下载5个东西,版本我都试好了,直接用:
#下载mmcv
mim install mmcv-full==1.6.0
#下载mmdet
mim install mmdet==2.27.0
#下载mmsegmentation
mim install mmsegmentation==0.30.0
#下载mmdet3d
mim install mmdet3d==1.0.0rc6
#顺便把可视化工具也下了,不然后面跑demo也省不掉
pip install open3d
然后就可以下载源码加编译了:
git clone https://github.com/open-mmlab/mmdetection3d.git
cd mmdetection3d
pip install -e .
到这里,就算完成了安装,可以简单验证一下,import不报错:
(det) xhao@zns:~$ python
Python 3.8.16 (default, Jan 17 2023, 23:13:24)
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import mmdet
>>> import mmdet3d
>>> exit()
后面有时间会讲验证,训练和测试。本人自己也在学习,一起进步哦~
-------------------------------------------------------------------
在训练的时候有个问题:AttributeError:module ‘distutils‘ has no attribute ‘version
这个时候去环境里看一下setuptools的版本,问题可能出在setuptools的版本过高。直接安装即可:
文章来源:https://www.toymoban.com/news/detail-433767.html
pip install setuptools==59.5.0 //需要比你之前的低
AttributeError:module ‘distutils‘ has no attribute ‘version_looseversion = distutils.version.looseversion attr_Zizizirrr的博客-CSDN博客文章来源地址https://www.toymoban.com/news/detail-433767.html
到了这里,关于mmdetection3d系列--(1)安装步骤(无坑版)的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!