Add Docker support

TensorFlow w/ GPU should be properly configured now.

Dockerfile

@@ -0,0 +1,18 @@
+FROM tensorflow/tensorflow:1.15.4-gpu-py3
+
+# Install dependencies
+ARG DEBIAN_FRONTEND=noninteractive
+RUN apt update && apt install -y \
+    libgl1-mesa-glx \
+    python3-pyqt5 \
+    python3-tk
+
+# Install Python dependencies
+WORKDIR /tmp
+COPY requirements.txt .
+RUN pip install -r requirements.txt
+
+# Add project code to PYTHONPATH (assumes volume mount point is /code)
+ENV PYTHONPATH="/code:$PYTHONPATH"
+
+WORKDIR /code

README_SUPPLEMENT.md

@@ -2,30 +2,36 @@
 
 Additional info (didn't want to modify original README).
 
-## Setup
+## Docker Usage
 
-### Installing dependencies
+### Build
 
 ```bash
-pip install -r requirements.txt
+docker build -t lanenet:latest .
 ```
 
-### Updating PYTHONPATH
+### Run
 
 ```bash
-export PYTHONPATH="/path/to/lanenet-lane-detection:$PYTHONPATH"
+docker run -it --rm \
+    --gpus all \
+    -u $(id -u):$(id -g) \
+    -e DISPLAY=$DISPLAY \
+    -v /tmp/.X11-unix/:/tmp/.X11-unix \
+    -v $PWD:/code \
+    lanenet:latest bash
 ```
 
 ## Testing LaneNet model
 
 ### Testing a single image
 
 ```bash
-python3 tools/test_lanenet.py --weights_path ./model/tusimple_lanenet/tusimple_lanenet.ckpt --image_path ./data/tusimple_test_image/0.jpg
+python tools/test_lanenet.py --weights_path ./model/tusimple_lanenet/tusimple_lanenet.ckpt --image_path ./data/tusimple_test_image/0.jpg
 ```
 
 ### Testing a video source
 
 ```bash
-python3 tools/test_lanenet_video.py --weights_path ./model/tusimple_lanenet/tusimple_lanenet.ckpt --video_src ./data/tusimple_test_video/0.mp4
+python tools/test_lanenet_video.py --weights_path ./model/tusimple_lanenet/tusimple_lanenet.ckpt --video_src ./data/tusimple_test_video/0.mp4
 ```

config/tusimple_lanenet.yaml

@@ -85,7 +85,7 @@ SOLVER:
         SCORE_THRESH: 0.65
         MIN_SAMPLE_NUMS: 65536
 GPU:
-    GPU_MEMORY_FRACTION: 0.9
+    GPU_MEMORY_FRACTION: 0.3
     TF_ALLOW_GROWTH: True
 POSTPROCESS:
     MIN_AREA_THRESHOLD: 100

requirements.txt

@@ -1,11 +1,11 @@
 easydict==1.9
 glog==0.3.1
 loguru==0.5.3
-numpy==1.18.5
+# numpy==1.18.5
 matplotlib==3.3.2
 opencv-python==4.4.0.44 
 PyYAML==5.3.1
 scipy==1.5.2
 scikit-learn==0.23.2
-tensorflow-gpu==1.15.4
+# tensorflow-gpu==1.15.4
 tqdm==4.49.0