一、拉取镜像[0]部署容器
从docker hub拉取mmdetection的镜像以建立容器:
docker search mmdetection
docker pull vistart/mmdetection
nvidia-docker run -it --shm-size='16g' --name='yshuqiao' -v /media/yzl/data1/gzx/data:/maps vistart/mmdetection /bin/bash
安装常用包:
apt install vim
apt install curl
在https://pypi.org/下载pip包并解压,在相应目录运行:
python setup.py install
修改mmdetection里面配置的图片路径、类别数、图片尺寸、imgs_per_gpu和学习率,下载预训练权重并配置训练路径,再在inference.py里设置测试集和权重路径,编写一个简单的run.sh(也放在mmdetection路径下)
python inference.py
二、用部署好的新容器commit镜像[1]
退出容器,然后:
docker commit bea719722d83 fabric:v1
三、编写Dockerfile
在宿主机的某个目录下新建一个空文件夹并创建Dockerfile:
mkdir builddocker
cd builddocker
vi Dockerfile
然后编写Dockerfile:
FROM yshuqiao/fabric: