yolov5如何运行自己的数据集_全网最细 | 教你如何在 docker 容器下使用 mmdetection 训练自己的数据集...

红色石头的个人网站：

红色石头的个人博客-机器学习、深度学习之路​www.redstonewill.com

这里不再介绍 mmdetection 的安装和配置，使用 mmdetection 较简单的方法是使用已安装 mmdetection 的 docker 容器。这样直接省去了安装 mmdetection 的过程，让重心放在模型训练上！

如果你对 docker 和 mmdetection 还不是很熟悉，请自行搜索一下，本文就不再赘述了。

这里附上 mmdetection 的 GitHub 地址：

https://github.com/open-mmlab/mmdetection

0. 前期准备

首先默认你的电脑已经做好了下面这些前期准备工作：

• Ubuntu 16.04 或以上
• GPU 安装
• cuda 安装
• cudnn 安装
• docker 安装
• nvidia-docker

当然，如果你连接的是公司或学校的服务器，且服务器已经做了上面几点准备，那你只需要一个 Xshell 远程登录服务器就行了。

1. 下载含 mmdetection 的 docker 镜像

首先，我们需要找到一个已经配置好 mmdetection 环境的 docker 镜像。可以在 dockerhub 上用 “mmdetection” 作为关键词进行搜索，也可以在 terminal 里直接使用命令 docker search 进行搜索：

$ 

结果显示如下图所示：

这里，我们选择排第一的 vistart/mmdetection 镜像，下载 docker 镜像的方法也很简单，使用 docker pull 从镜像仓库中拉取指定镜像：

$ 

如果网络没问题，下载会在几分钟之内完成。下载完成之后，我们就可以查看 vistart/mmdetection 镜像是否已经放在本地镜像里了：

$ 

可以看到 vistart/mmdetection 镜像已经成功下载了