mohamed-boudjoghra commited on
Commit
5f3418a
β€’
1 Parent(s): 2b657e9

Upload 4 files

Browse files
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