Added Docker Utils For Quick Env Setup

.gitignore

@@ -0,0 +1,3 @@
+__pycache__/*
+outputs/*
+runs/*

README.md

@@ -8,6 +8,7 @@
 
 ## 🆕 Update Log
 
+- **[2025.07.07]** Now **docker** scripts available!  
 - **[2025.04.23]** We have released the **2rd stage training code**! 🎉  
 - **[2025.04.11]** We have released the **inference code**! 🎉  
 
@@ -36,6 +37,39 @@ DMD³C introduces a novel framework for **fine-grained depth completion** by dis
 
 ## 🚀 Getting Started
 
+### (Docker only)
+__Clone repo and set up docker:__
+```sh
+git clone https://github.com/fetty31/DMD3C.git # clone repo
+cd DMD3C/docker/
+chmod +x build run
+./build # build docker image
+./run   # run docker container
+```
+
+_Note that inside the `run` script the directory on your PC with the KITTI data is shared with the docker container via:_
+```sh
+--volume="$HOME/Desktop/kitti_dataset:/home/kitti_dataset" 
+```
+_Change this line (80) inside the [run](docker/run) script in order to match with your KITTI dataset parent folder._
+
+__Once inside docker, set up environment:__
+```sh
+## 🐳 Inside the Docker container
+# build conda env
+cd home/DMD3C/
+conda env create -f environment.yml 
+conda init && source /root/.bashrc
+conda activate bp
+
+# build cuda extension
+cd exts
+export CC=/usr/bin/gcc
+export CXX=/usr/bin/g++
+python setup.py install 
+```
+__From this, jump to steps 3-5__
+
 ### 1. Clone Base Repository
 
 ```bash
@@ -65,7 +99,14 @@ Open `demo.py` and go to **line 338**, where you can modify the input sequence p
 
 ```python
 # demo.py (Line 338)
-sequence = "/path/to/your/kitti/sequence"
+base = "/your/path/to/kitti/sequence"
+```
+
+_(Or if you are using docker):_
+```python
+# demo_docker.py (Line 418)
+global_path = "/your/path/to/kitti/"
+base = global_path + "/sequence"
 ```
 
 Download pre-trained weights:
@@ -80,6 +121,11 @@ Run inference:
 bash demo.sh
 ```
 
+_(Or if you are using docker)_
+```bash
+bash demo_docker.sh
+```
+
 You will get results like this:
 
 ![supp-video 00_00_00-00_00_30](https://github.com/user-attachments/assets/a1412bca-c368-4d19-a081-79eeabaa2901)

demo.py

@@ -335,7 +335,6 @@ def main(cfg):
         net = run.net_ema.module.cuda()
         net.eval()
         # 读取左目图像
-        # base = "datas/kitti/raw/2011_09_26/2011_09_26_drive_0002_sync"
         base = "datas/kitti/raw/2011_09_26/2011_09_26_drive_0048_sync"
 
         image_type = 'color'  # 'grayscale' or 'color' image