环境配置(Linux)
- 创建虚拟环境：
conda create -n 2023_CVPR_VirConvNet python=3.9.13 - 激活虚拟环境：
conda activate 2023_CVPR_VirConvNet - 安装pytorch：
pip3 install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html - 测试torch：
pythonimport torchtorch.cuda.is_available()exit()
- 安装spconv：
pip install spconv-cu118pip install spconv-cu111 - 测试spconv：
pythonimport spconvexit()
- 进入代码根目录：
- 安装Numba：
pip install Numba==0.53.1 - 安装numpy1.23.0：
pip uninstall numpypip install numpy==1.23.0
- 安装pyyaml：
pip install pyyaml - 安装easydict：
pip install easydict - 安装scikit-image：
pip install scikit-image - 安装tqdm：
pip install tqdm - 安装tensorboardX：
pip install tensorboardX - 安装prefetch_generator：
pip install prefetch_generator - 安装opencv：
pip install opencv-python - 安装llvmlite：
pip install llvmlite - 安装SharedArray：
pip install SharedArray - 安装pyquaternion：
pip install pyquaternion - 运行pcdet安装程序：
python setup.py develop - setup pointnet：进入detr_third_party中有setup的地方，执行：
python setup.py install - setup MinkowskiEngine:
conda install cudatoolkit=11.1 -c pytorch -c nvidiaconda install openblas-devel -c anacondapip install ninja- 进入MinkowskiEngine工程文件夹执行：
python setup.py install --blas_include_dirs=${CONDA_PREFIX}/include --blas=openblas
数据配置
- 下载全部数据：| kitti |baidu (japc) /74GB| Location: …/data/kitti|
- 下载已训练好的模型：
- | Spconv2.1 | VirConv-L | baidu(k2dp) / 51M| Location: …/VirConv-L2.pth |
- | Spconv2.1 | VirConv-T | baidu(a4r4) / 56M| Location: …/VirConv-T2.pth |
- | Spconv2.1 | VirConv-S | baidu(j3mi) / 56M| Location: …/VirConv-S2.pth |
- | PENet | baidu (gp68) / 50M| Location: …/tools/PENet/pe.pth.tar |
运行
- 进入tools文件夹：
cd tools - 检查GPU：
nvidia-smi - 输入测试命令：
CUDA_VISIBLE_DEVICES=6 python3 test.py --cfg_file cfgs/models/kitti/VirConv-S.yaml --ckpt VirConv-S2.pth
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
