上交大-人工智能学院-陈思衡组(MAGIC)
飞书用户2119
添加快捷方式
分享
CoAlign Installation Guide [English Ver.]
输入“/”快速插入内容
CoAlign Installation Guide [English Ver.]
Author
: Yifan Lu
yifan_lu@sjtu.edu.cn
Installation
All operations should be done
on machines with GPUs
.
1.
Install
spconv 1.2.1 or 2.x
•
To
install
spconv 2.x,run:
代码块
Bash
# (if you are using cuda 11.3):
pip install spconv-cu113
# and install cumm and timm
pip install cumm
pip install timm
It should be noted that the checkpoint provided is trained by spconv 1.2.1, and the format in which the weights are saved is not interoperable.
•
To
install
spconv 1.2.1,please refer to:
https://github.com/traveller59/spconv/tree/v1.2.1.
2.
Install CoAlign
代码块
PowerShell
git clone https://github.com/yifanlu0227/CoAlign.git
cd CoAlign
pip install -r requirements.txt
python setup.py develop
# Bbx IOU cuda version compile
python opencood/utils/setup.py build_ext --inplace
# FPVRCNN dependency
python opencood/pcdet_utils/setup.py build_ext --inplace
These steps are the same as
https://opencood.readthedocs.io/en/latest/md_files/installation.html
3.
Install pypcd: required by DAIR-V2X LiDAR dataloader
Note that pypcd pip installing is not compatible with Python3. Therefore
a modified version
should be manually installed as followings.
代码块
Bash
# go to another folder. Do not clone it within CoAlign
git clone https://github.com/klintan/pypcd.git
cd pypcd
pip install python-lzf
python setup.py install
4.
Install EfficinetNet: required by camera detector Lift-Splat-Shoot
代码块
PowerShell
pip install efficientnet_pytorch==0.7.0
5.
Install some packages
代码块
PowerShell
pip install icecream
pip install opencv-python-headless
pip install open3d
pip install matplotlib
pip install tensorboardx
pip install pyyaml
pip install h5py
pip install einops
pip install pyquaternion
mmcv can not be installed, and it can be commented out when encountering 'import mmcv'
or install the mmcv by
pip install openmim
mim install mmcv-full==1.4.0