pc

Docker Usage

Build

docker build -t lanenet:latest .

Run

docker run -it --rm \
    --gpus all \
    -p 6006:6006 \
    -u $(id -u):$(id -g) \
    -e DISPLAY=$DISPLAY \
    -v /tmp/.X11-unix/:/tmp/.X11-unix \
    -v $PWD/..:/code \
    lanenet:latest

Testing Pre-trained TensorFlow LaneNet model

NOTE: It may take awhile to start up, but once it does, inference is pretty quick.

Testing a single image

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

python tools/test_lanenet_video.py --weights_path ./model/tusimple_lanenet/tusimple_lanenet.ckpt --video_src ./data/tusimple_test_video/0.mp4