该内容主要介绍了通过yolov5 docker镜像创建容器,以及对自己的数据集进行简单的训练和推理。使用镜像可以省去很多安装运行环境、依赖失败的问题。需要本地已安装部署了docker服务,并且对网络环境有一定的要求。
- 查询yolov5镜像
docker search yolov5
这里选择使用 ultralytics/yolov5
- 拉取镜像
- 如果需要拉取某一版本的镜像,则在镜像名称后面用冒号并连接版本号,比如ultralytics/yolov5:1.0.0
docker pull ultralytics/yolov5
- 运行镜像,创建容器
- 创建容器时,建议将宿主机的某个目录挂载到容器的某个目录下,这样使用容器中的yolov5算法进行训练时,方便获取宿主机中的数据
docker run -it --name=yolov5_service -d -v C:\root\source-code\projects\