win10 mmdetection3D 安装

Posted 2022-08-31 东东就是我

经过了2天的折腾，发现更换阿里源是最简单的方式

1.换源

在C:\\Users******.condarc 修改为阿里源

show_channel_urls: true
default_channels:
  - http://mirrors.aliyun.com/anaconda/pkgs/main
  - http://mirrors.aliyun.com/anaconda/pkgs/r
  - http://mirrors.aliyun.com/anaconda/pkgs/msys2
custom_channels:
  conda-forge: http://mirrors.aliyun.com/anaconda/cloud
  msys2: http://mirrors.aliyun.com/anaconda/cloud
  bioconda: http://mirrors.aliyun.com/anaconda/cloud
  menpo: http://mirrors.aliyun.com/anaconda/cloud
  pytorch: http://mirrors.aliyun.com/anaconda/cloud
  simpleitk: http://mirrors.aliyun.com/anaconda/cloud

2. 配置环境

https://blog.csdn.net/m0_68312479/article/details/126165418

conda create -n open-mmlab python==3.7
conda activate open-mmlab
conda install pytorch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1 cudatoolkit=10.2 -c pytorch
pip3 install openmim
mim install mmcv-full
mim install mmdet
mim install mmsegmentation
git clone https://github.com/open-mmlab/mmdetection3d.git
cd mmdetection3d
pip3 install -e .

//安装spconv需加-cu102注意对应自己的cuda版本
pip install spconv-cu102

3.测试

1. 下载模型

https://github.com/open-mmlab/mmdetection3d/blob/master/configs/second/hv_second_secfpn_6x8_80e_kitti-3d-3class.py

2. 运行代码

python demo/pcd_demo.py demo/data/kitti/kitti_000008.bin configs/second/hv_second_secfpn_6x8_80e_kitti-3d-car.py checkpoints\\hv_second_secfpn_6x8_80e_kitti-3d-3class_20210831_022017-ae782e87.pth --out-dir data/output_result

4.训练

下载数据

https://blog.csdn.net/m0_68312479/article/details/126165382