当前位置:   article > 正文

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

mmdetection3d安装

  最近在看一些基于点云3d目标检测的文章,需要复现甚至修改一些算法,就找到了mmlab开源的mmdetection3d目标检测框架,方便后续学习。

    在安装的时候遇到一点坑,比如环境问题,安装完能跑demo但是不能跑训练测试问题等。在解决问题后还是完成了安装。在这里记录一下正确的安装流程,已备再次查阅,也给大家提供一点参考。

    首先建环境,最好是新建一个环境

  1. conda create -n det python=3.8
  2. #激活
  3. 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个东西,版本我都试好了,直接用:

  1. #下载mmcv
  2. mim install mmcv-full==1.6.0
  3. #下载mmdet
  4. mim install mmdet==2.27.0
  5. #下载mmsegmentation
  6. mim install mmsegmentation==0.30.0
  7. #下载mmdet3d
  8. mim install mmdet3d==1.0.0rc6
  9. #顺便把可视化工具也下了,不然后面跑demo也省不掉
  10. pip install open3d

    然后就可以下载源码加编译了:

  1. git clone https://github.com/open-mmlab/mmdetection3d.git
  2. cd mmdetection3d
  3. pip install -e .

    到这里,就算完成了安装,可以简单验证一下,import不报错:

  1. (det) xhao@zns:~$ python
  2. Python 3.8.16 (default, Jan 17 2023, 23:13:24)
  3. [GCC 11.2.0] :: Anaconda, Inc. on linux
  4. Type "help", "copyright", "credits" or "license" for more information.
  5. >>> import mmdet
  6. >>> import mmdet3d
  7. >>> exit()

后面有时间会讲验证,训练和测试。本人自己也在学习,一起进步哦~

-------------------------------------------------------------------

    在训练的时候有个问题:AttributeError:module ‘distutils‘ has no attribute ‘version

这个时候去环境里看一下setuptools的版本,问题可能出在setuptools的版本过高。直接安装即可:
 

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

AttributeError:module ‘distutils‘ has no attribute ‘version_looseversion = distutils.version.looseversion attr_Zizizirrr的博客-CSDN博客

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/很楠不爱3/article/detail/110125
推荐阅读
相关标签
  

闽ICP备14008679号