Skip to content

Instantly share code, notes, and snippets.

@ck196
Last active March 30, 2017 08:00
Show Gist options
  • Save ck196/3660efe9e0ceab1b0a4ad5f01aaef923 to your computer and use it in GitHub Desktop.
Save ck196/3660efe9e0ceab1b0a4ad5f01aaef923 to your computer and use it in GitHub Desktop.

Yêu cầu

  • Ubuntu >= 14.04, GPU( >= 2GB), CUDA 7.5+, cuDnn 4.0.7+
  • Install docker, docker-nvidia

Cài đặt

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

Chú ý

  • 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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment