相关环境准备
在官网下载docker
启动容器、加载镜像、挂载数据集路径
#拉取镜像
sudo docker pull ultralytics/yolov5:latest
docker run --name yolov5_test --ipc=host -it ultralytics/yolov5:latest
#可以指定端口映射进行启动
docker run -v /home/lbh/ubuntu-home/datasets --name yolov5_test -p 16022:22 -it yolov5_require:torch1.8-cu102-runtime-py3.9-ub18
#指定端口映射、指定GPU模式进行启动
sudo docker run -v --runtime=nvidia /home/lbh/yolov5_v6.1_test:/tmp/project -p 16022:22 --name yolov5_gpu --gpus all -it ultralytics/yolov5:latest
XXX
## 停止容器
docker stop yolov5_test
## 关闭容器
docker kill yolov5_test
## 重启容器
docker restart yolov5_test
## 删除容器
docker rm yolov5_test
# 运行检测代码
python detect.py --weights yolov5s.pt --source uniform/data/images
# 将检测结果从容器拷贝到宿主机
docker cp myyolov5:/usr/src/app/runs/detect D:\datasets\uniform\runs