cd /content/drive/MyDrive

 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

你可能感兴趣的:(深度学习,pytorch)