cd /content/drive/MyDrive/GD-MAE
python setup.py develop --user
cd pcdet/ops/dcn
rm -rf build/
python setup.py develop --user
python train.py --cfg_file ./cfgs/kitti_models/pointpillar.yaml
Waymo Dataset — MMDetection3D 1.3.0 documentation
https://github.com/open-mmlab/mmdetection3d/blob/dev-1.x/docs/en/user_guides/dataset_prepare.md#summary-of-annotation-files
https://console.cloud.google.com/storage/browser/waymo_open_dataset_v_1_2_0/training?pageState=(%22StorageObjectListTable%22:(%22f%22:%22%255B%255D%22))&prefix=&forceOnObjectsSortingFiltering=false
failed call to cuInit: CUDA_ERROR_NOT_INITIALIZED: initialization error
CUDA_VISIBLE_DEVICES=0,1,2 tools/dist_train.sh configs/pointpillars/pointpillars_hv_fpn_sbn-all_8xb4-2x_nus-3d.py 3
./tools/dist_train.sh configs/pointpillars/pointpillars_hv_fpn_sbn-all_8xb4-2x_nus-3d.py
#unzip "/content/drive/MyDrive/GD-MAE-main/data/kitti/data_object_calib.zip" -d "/content/drive/MyDrive/GD-MAE-main/data/kitti"
#unzip "/content/drive/MyDrive/GD-MAE-main/data/kitti/data_object_image_2.zip" -d "/content/drive/MyDrive/GD-MAE-main/data/kitti"
#unzip "/content/drive/MyDrive/GD-MAE-main/data/kitti/data_object_label_2.zip" -d "/content/drive/MyDrive/GD-MAE-main/data/kitti"
unzip "/content/drive/MyDrive/GD-MAE-main/data/kitti/data_object_velodyne.zip" -d "/content/drive/MyDrive/GD-MAE-main/data/kitti"
cd /content/drive/MyDrive/GD-MAE-main
python mmdetection3d/tools/create_data.py kitti --root-path ./data/kitti --out-dir ./data/kitti --extra-tag kitti
python -m pcdet.datasets.kitti.kitti_dataset create_kitti_infos tools/cfgs/dataset_configs/kitti_dataset.yaml
通过百度网盘分享的文件:20231213…
链接:https://pan.baidu.com/s/11FVpcRRtQC4Db4gKPRQfzQ?pwd=rckh
提取码:rckh
复制这段内容打开「百度网盘APP 即可获取」
cd /content/drive/MyDrive/GD-MAE-main/data/kitti
wget https://s3.eu-central-1.amazonaws.com/avg-kitti/data_object_image_2.zip
wget https://s3.eu-central-1.amazonaws.com/avg-kitti/data_object_label_2.zip
wget https://s3.eu-central-1.amazonaws.com/avg-kitti/data_object_velodyne.zip
wget https://s3.eu-central-1.amazonaws.com/avg-kitti/data_object_calib.zip
%%shell
cd /content/drive/MyDrive
conda create -n gd-mae python=3.7
#conda upgrade -n base -c defaults --override-channels conda
conda update -n base -c defaults conda
#conda init
#conda deactivate
activate gd-mae
conda install -y pytorch==1.10.1 torchvision==0.11.2 cudatoolkit=11.1 -c pytorch -c conda-forge
conda install -y -c fvcore -c iopath -c conda-forge fvcore iopath
conda install -y pytorch3d -c pytorch3d
pip install numpy==1.19.5 protobuf==3.19.4 scikit-image==0.19.2 waymo-open-dataset-tf-2-2-0 nuscenes-devkit==1.0.5 spconv-cu111 numba scipy pyyaml easydict fire tqdm shapely matplotlib opencv-python addict pyquaternion awscli open3d pandas future pybind11 tensorboardX tensorboard Cython
pip install torch-scatter -f https://data.pyg.org/whl/torch-1.10.1+cu111.html
#git clone https://github.com/Nightmare-n/GD-MAE
conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.7 -c pytorch -c nvidia
pip install torch==2.0.0+cu117 torchvision==0.15.1+cu117 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu117
pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu113/torch_stable.html
https://github.com/BIT-DYN/pointpillars_ros.git
%%shell
cd /content/drive/MyDrive/GD-MAE-main
conda create -n openmmlab python=3.7 -y
activate openmmlab
pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
pip3 install openmim
mim install mmcv-full
# 安装 mmdetection
pip install git+https://github.com/open-mmlab/mmdetection.git
# 安装 mmsegmentation
pip install git+https://github.com/open-mmlab/mmsegmentation.git
# 安装 mmdetection3d
git clone https://github.com/open-mmlab/mmdetection3d.git
cd mmdetection3d
pip install -v -e .
pip install mmengine
#pip uninstall MMCV
#pip install MMCV==2.0.0