mohamed-boudjoghra
commited on
Commit
β’
5f3418a
1
Parent(s):
2b657e9
Upload 4 files
Browse files- docs/Data_prep.md +64 -0
- docs/Installation.md +64 -0
- docs/pipeline.png +0 -0
- docs/qualitatives.png +0 -0
docs/Data_prep.md
ADDED
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
## ScanNet200
|
2 |
+
Download <a href="https://kaldir.vc.in.tum.de/scannet_benchmark/documentation">ScaNnet200</a> and follow <a href="[https://kaldir.vc.in.tum.de/scannet_benchmark/documentation](https://github.com/JonasSchult/Mask3D)">Mask3D</a> to preprocess data. A preprocessed point cloud scene has the following format `id.npy`. Kindly move all preprocessed scenes to their corresponding folders e.g. `0011_00.npy` to `./data/scannet200/scene0011_00`.
|
3 |
+
|
4 |
+
ScanNet200 data folder should have the following structure
|
5 |
+
```
|
6 |
+
./data
|
7 |
+
βββ scannet200
|
8 |
+
βββ ground_truth
|
9 |
+
βββ scene0011_00.txt
|
10 |
+
βββ ...
|
11 |
+
βββ scene0011_00
|
12 |
+
βββ poses <- folder with camera poses
|
13 |
+
β βββ 0.txt
|
14 |
+
β βββ 1.txt
|
15 |
+
β βββ ...
|
16 |
+
βββ color <- folder with RGB images
|
17 |
+
β βββ 0.jpg (or .png/.jpeg)
|
18 |
+
β βββ 1.jpg (or .png/.jpeg)
|
19 |
+
β βββ ...
|
20 |
+
βββ depth <- folder with depth images
|
21 |
+
β βββ 0.png (or .jpg/.jpeg)
|
22 |
+
β βββ 1.png (or .jpg/.jpeg)
|
23 |
+
β βββ ...
|
24 |
+
βββ intrinsics.txt <- camera intrinsics
|
25 |
+
β
|
26 |
+
βββ 0011_00.npy <- preprocessed point cloud of the scene
|
27 |
+
β
|
28 |
+
βββ scene0011_00_vh_clean_2.ply <- raw point cloud of the scene
|
29 |
+
βββ ...
|
30 |
+
```
|
31 |
+
|
32 |
+
## Replica
|
33 |
+
|
34 |
+
Run the following command.
|
35 |
+
```
|
36 |
+
sh scripts/get_replica_dataset.sh
|
37 |
+
```
|
38 |
+
|
39 |
+
A replica folder with the following format will be generated.
|
40 |
+
```
|
41 |
+
./data
|
42 |
+
βββ replica
|
43 |
+
βββ ground_truth
|
44 |
+
βββ office0.txt
|
45 |
+
βββ ...
|
46 |
+
βββ office0
|
47 |
+
βββ poses <- folder with camera poses
|
48 |
+
β βββ 0.txt
|
49 |
+
β βββ 1.txt
|
50 |
+
β βββ ...
|
51 |
+
βββ color <- folder with RGB images
|
52 |
+
β βββ 0.jpg (or .png/.jpeg)
|
53 |
+
β βββ 1.jpg (or .png/.jpeg)
|
54 |
+
β βββ ...
|
55 |
+
βββ depth <- folder with depth images
|
56 |
+
β βββ 0.png (or .jpg/.jpeg)
|
57 |
+
β βββ 1.png (or .jpg/.jpeg)
|
58 |
+
β βββ ...
|
59 |
+
βββ intrinsics.txt <- camera intrinsics
|
60 |
+
β
|
61 |
+
βββ office0_mesh.ply <- raw point cloud of the scene
|
62 |
+
βββ ...
|
63 |
+
```
|
64 |
+
|
docs/Installation.md
ADDED
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
## Clone the repository
|
2 |
+
```
|
3 |
+
git clone https://github.com/aminebdj/OpenYOLO3D.git
|
4 |
+
cd OpenYOLO3D
|
5 |
+
|
6 |
+
```
|
7 |
+
|
8 |
+
## Conda Environment
|
9 |
+
```
|
10 |
+
export TORCH_CUDA_ARCH_LIST="6.0 6.1 6.2 7.0 7.2 7.5 8.0 8.6"
|
11 |
+
|
12 |
+
conda env create -f environment.yml
|
13 |
+
|
14 |
+
conda activate openyolo3d
|
15 |
+
|
16 |
+
cd models/Mask3D/
|
17 |
+
pip3 install torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113
|
18 |
+
pip3 install torch-scatter -f https://data.pyg.org/whl/torch-1.12.1+cu113.html
|
19 |
+
pip3 install 'git+https://github.com/facebookresearch/detectron2.git@710e7795d0eeadf9def0e7ef957eea13532e34cf' --no-deps
|
20 |
+
|
21 |
+
cd third_party
|
22 |
+
|
23 |
+
git clone --recursive "https://github.com/NVIDIA/MinkowskiEngine"
|
24 |
+
cd MinkowskiEngine
|
25 |
+
git checkout 02fc608bea4c0549b0a7b00ca1bf15dee4a0b228
|
26 |
+
python setup.py install --force_cuda --blas=openblas
|
27 |
+
|
28 |
+
cd ..
|
29 |
+
git clone https://github.com/ScanNet/ScanNet.git
|
30 |
+
cd ScanNet/Segmentator
|
31 |
+
git checkout 3e5726500896748521a6ceb81271b0f5b2c0e7d2
|
32 |
+
make
|
33 |
+
cd ../../
|
34 |
+
|
35 |
+
cd pointnet2
|
36 |
+
python setup.py install
|
37 |
+
|
38 |
+
cd ../../
|
39 |
+
pip3 install pytorch-lightning==1.7.2
|
40 |
+
|
41 |
+
pip install black==21.4b2 cloudpickle==3.0.0 future hydra-core==1.0.5 pycocotools>=2.0.2 pydot iopath==0.1.7 loguru albumentations
|
42 |
+
|
43 |
+
pip install .
|
44 |
+
|
45 |
+
cd ../YOLO-World
|
46 |
+
pip install -e .
|
47 |
+
cd ../../
|
48 |
+
pip install mmyolo==0.6.0 mmdet==3.0.0 plyfile
|
49 |
+
pip install openmim
|
50 |
+
pip3 install torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113
|
51 |
+
mim install mmcv==2.0.0
|
52 |
+
pip install open3d
|
53 |
+
pip install pillow==9.1.0
|
54 |
+
|
55 |
+
|
56 |
+
```
|
57 |
+
## Prepare the class agnostic masks
|
58 |
+
```
|
59 |
+
sh scripts/get_class_agn_masks.sh
|
60 |
+
```
|
61 |
+
## Prepare the checkpoints
|
62 |
+
```
|
63 |
+
sh scripts/get_checkpoints.sh
|
64 |
+
```
|
docs/pipeline.png
ADDED
docs/qualitatives.png
ADDED