Docker-开源Detectron(补充)

nvidia-docker在之前创建的镜像detectron下使用本地文件

1..git cloneDetectron,解压后存放路径/home/pandamax/Detectron
2.在文件/home/pandamax/Detectron/lib/datasets/data下创建coco文件夹,用来存放对应的数据,coco 文件夹格式为:

coco
|_ coco_train2014
|  |_ <im-1-name>.jpg
|  |_ ...
|  |_ <im-N-name>.jpg
|_ coco_val2014
|_ ...
|_ annotations
   |_ instances_train2014.json
   |_ ...

3.确定镜像detectron的镜像ID,操作示例:

• 终端输入 sudo docker images,查看image ID
  

4.挂载本地文件

sudo nvidia-docker run -it -v /home/pandamax/Detectron:/detectron 128ed0b4a26b

5.编译caffe2

cd /path/to/lib
make
make ops

6.Training a Model with Detectron

python2 tools/train_net.py \
    --cfg configs/getting_started/tutorial_1gpu_e2e_faster_rcnn_R-50-FPN.yaml \
    OUTPUT_DIR /tmp/detectron-output