Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

AG_Cylider3D #154

Draft
wants to merge 9 commits into
base: master
Choose a base branch
from
Draft

AG_Cylider3D #154

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
058b3c1
update to latest version of modules
HiImShynn Sep 18, 2022
5a7818d
add attention_block_3d.py
HiImShynn Sep 23, 2022
95d77fb
add attention_unet_3d.py
HiImShynn Sep 23, 2022
452924f
refine some old codes
HiImShynn Oct 27, 2022
387c940
change version='v1.0-trainval' to 'v1.0-mini'
HiImShynn Oct 27, 2022
7bd15ec
change 'nuScenes' to 'nuscenes' in imageset path
HiImShynn Oct 28, 2022
40d5b8f
fix nuScenes.yaml config
HiImShynn Oct 31, 2022
05e2efb
fix stride in W_x
HiImShynn Nov 2, 2022
1f7557b
update Attention_gate
HiImShynn Dec 13, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions .gitignore
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
# Created by .ignore support plugin (hsz.mobi)
.idea
# Created by .ignore support plugin (hsz.mobi)
.idea
.DS_Store
114 changes: 57 additions & 57 deletions Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,57 +1,57 @@
FROM nvidia/cuda:10.2-cudnn8-devel-ubuntu18.04

# Just in case we need it
ENV DEBIAN_FRONTEND noninteractive

RUN apt update && apt upgrade -y
RUN apt install -y python3 python3-dev python3-pip
RUN pip3 install --upgrade pip
RUN apt install -y git wget unzip
RUN apt install -y build-essential
RUN apt install -y ffmpeg libsm6 libxext6

# get newer version of cmake
RUN wget https://apt.kitware.com/kitware-archive.sh
RUN chmod +x kitware-archive.sh && ./kitware-archive.sh

RUN pip3 install torch==1.6.0
RUN pip3 install pyyaml==5.4.1
RUN pip3 install cython==0.29.24
RUN pip3 install nuscenes-devkit==1.1.6
RUN pip3 install numba==0.53.1
RUN pip3 install strictyaml==1.4.4

# Torch scatter
RUN pip3 install --no-index torch-scatter -f https://pytorch-geometric.com/whl/torch-1.6.0+cu102.html

# SpConv
# need to use SpConv 1 and hotfixes
# see: https://github.com/traveller59/spconv/issues/58
RUN git clone --recursive https://github.com/traveller59/spconv.git /spconv
RUN wget -O hotfixes.zip https://github.com/traveller59/spconv/files/4658204/hotfixes.zip
RUN unzip hotfixes -d /hotfixes
WORKDIR /spconv
RUN git checkout 8da6f967fb9a054d8870c3515b1b44eca2103634
# needs to be done before we can apply the patches
RUN git config --global user.email "[email protected]"
RUN git config --global user.name "Test"
RUN git am /hotfixes/0001-fix-problem-with-torch-1.4.patch
RUN git am /hotfixes/0001-Allow-to-specifiy-CUDA_ROOT-directory-and-pick-corre.patch
RUN apt install -y libboost-all-dev cmake
ENV CUDA_ROOT=/usr/local/cuda
RUN pip3 install wheel
WORKDIR /spconv/third_party
RUN git clone https://github.com/pybind/pybind11.git
WORKDIR /spconv
RUN python3 setup.py bdist_wheel
WORKDIR /spconv/dist
RUN pip3 install *.whl

## Install Cylinder 3D
WORKDIR /
RUN git clone --recursive --depth 1 https://github.com/xinge008/Cylinder3D
WORKDIR /Cylinder3D
RUN chmod +x train_nusc.sh
RUN mkdir -p /data/dataset/nuScenes


FROM nvidia/cuda:10.2-cudnn8-devel-ubuntu18.04
# Just in case we need it
ENV DEBIAN_FRONTEND noninteractive
RUN apt update && apt upgrade -y
RUN apt install -y python3 python3-dev python3-pip
RUN pip3 install --upgrade pip
RUN apt install -y git wget unzip
RUN apt install -y build-essential
RUN apt install -y ffmpeg libsm6 libxext6
# get newer version of cmake
RUN wget https://apt.kitware.com/kitware-archive.sh
RUN chmod +x kitware-archive.sh && ./kitware-archive.sh
RUN pip3 install torch==1.6.0
RUN pip3 install pyyaml==5.4.1
RUN pip3 install cython==0.29.24
RUN pip3 install nuscenes-devkit==1.1.6
RUN pip3 install numba==0.53.1
RUN pip3 install strictyaml==1.4.4
# Torch scatter
RUN pip3 install --no-index torch-scatter -f https://pytorch-geometric.com/whl/torch-1.6.0+cu102.html
# SpConv
# need to use SpConv 1 and hotfixes
# see: https://github.com/traveller59/spconv/issues/58
RUN git clone --recursive https://github.com/traveller59/spconv.git /spconv
RUN wget -O hotfixes.zip https://github.com/traveller59/spconv/files/4658204/hotfixes.zip
RUN unzip hotfixes -d /hotfixes
WORKDIR /spconv
RUN git checkout 8da6f967fb9a054d8870c3515b1b44eca2103634
# needs to be done before we can apply the patches
RUN git config --global user.email "[email protected]"
RUN git config --global user.name "Test"
RUN git am /hotfixes/0001-fix-problem-with-torch-1.4.patch
RUN git am /hotfixes/0001-Allow-to-specifiy-CUDA_ROOT-directory-and-pick-corre.patch
RUN apt install -y libboost-all-dev cmake
ENV CUDA_ROOT=/usr/local/cuda
RUN pip3 install wheel
WORKDIR /spconv/third_party
RUN git clone https://github.com/pybind/pybind11.git
WORKDIR /spconv
RUN python3 setup.py bdist_wheel
WORKDIR /spconv/dist
RUN pip3 install *.whl
## Install Cylinder 3D
WORKDIR /
RUN git clone --recursive --depth 1 https://github.com/xinge008/Cylinder3D
WORKDIR /Cylinder3D
RUN chmod +x train_nusc.sh
RUN mkdir -p /data/dataset/nuScenes
Loading