- Ubuntu >= 14.04, GPU( >= 2GB), CUDA 7.5+, cuDnn 4.0.7+
- Install docker, docker-nvidia
docker pull ck196/py-faster-rcnn
Download pre-trained model. https://drive.google.com/open?id=0B4MXf_cfl_weOEMyTHVSRWF6M0U
- Đặt file download vào /data/. đặt tên file là 540k.caffemodel. Thư mục /data/ là thư mục share giữa host (máy thật) và container (máy ảo)
- Chạy docker container. (với webcam). Nếu không có webcam thì sửa /root/py-faster-rcnn/tools/detect-agen-demo.py để phù hợp với cách chạy.
#!/bin/bash
xhost +
GPU=0 nvidia-docker run --privileged --rm -it \
--env DISPLAY=$DISPLAY \
--env PY_FASTER_RCNN_PATH=/root/py-faster-rcnn \
--env="QT_X11_NO_MITSHM=1" \
-w /root/py-faster-rcnn \
-v /dev/video0:/dev/video0 \
-v /tmp/.X11-unix:/tmp/.X11-unix:ro \
-v /data:/data:rw \
ck196/py-faster-rcnn python tools/detect-agen-demo.py
xhost -
- Hoặc khi không có webcam muốn vào chỉnh sửa.
nvidia-docker run --rm -it -v /data:/data:rw ck196/py-faster-rcnn bash
- Mỗi lần chạy container khi tắt đi sẽ ko lưu lại các thứ chỉnh sửa.
- Muốn lưu lại cần thực commit container hiện tại vào một image mới.
docker commit container_id new_image_name