当前位置: 代码网 > 科技>人工智能 > mmdetection3d系列--(1)安装步骤(无坑版)

mmdetection3d系列--(1)安装步骤(无坑版)

2024年07月28日 人工智能 我要评论
mmdetecion3d无坑版安装教程。在这里记录一下正确的安装流程,已备再次查阅,也给大家提供一点参考。

  最近在看一些基于点云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的版本过高。直接安装即可:
 

pip install setuptools==59.5.0 //需要比你之前的低
(0)

相关文章:

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论

验证码:
Copyright © 2017-2025  代码网 保留所有权利. 粤ICP备2024248653号
站长QQ:2386932994 | 联系邮箱:2386932994@qq.com