MMlab简介
MMLab是基于Pytorch搭建的人工智能模型库,内部包含了计算机视觉领域的很多研究方向,我是做目标检测的,所以我主要关注的是MMdetection、MMRotate、MMYolo这三部分系列。MMdetection是做水平框目标检测的,MMRotate是做旋转框目标检测的,这是一个比较小众的方向,MMYolo整合了YOLO系列的算法,从Yolo-v5到Yolo-v8都可以自由使用。
MMdetection简介
我主要做目标检测的,所以介绍一下MMdetection这个模型库。MMdetection整合了经典算法与前沿算法,它不仅是一个模型库,还可以作为入门目标检测领域的指导教程,我们可以根据提供的模型去阅读原论文,这样有论文,有代码,我们会进步神速。link,这里是MMdetection的Github地址,可以去点点star。
如何安装MMdetection
对于新手来说,安装环境是一件很令人头痛的事情,我自己在刚开始的时候也遇到了很多问题,所以我把自己的安装环境路线写在这里,分享给大家,希望能够帮助大家避坑。
系统选择
雖然最近Mmdetection可以安裝在Windows操作系統下面,但是我們還是建議在Linux系統下面進行安裝Mmdetection.可以選擇Ubuntu18.04或者Ubuntu20.04的版本都可以.如果下夥伴不會安裝,我們下面給出來鏈接,個人選擇的是安裝雙系統,沒有選擇安裝虛擬機.
Ubuntu18.04安裝
Cuda与Cudnn安装
这里需要说明在安装完Cuda之后,无比要安装Cudnn,他是对显卡计算起到加速的作用.Cuda与Cudnn安装
Anaconda安装
Anaconda提供了各种编程需要的库,下载起来很方便.Anaconda自带Python,不需要在安装Python.Anaconda安装
Pycharm安装
Pycharm是Python的IDE,当然还有其他的IDE.例如VScode.Pycharm安装
安装Pytorch
Mmdetection是对Pytorch的二次封装,所以在安装Mmdetection需要提前安装Pytorch,Pytorch版本与Cuda版本有着对应关系,见下图.
Mmdetection与Mmcv版本对应关系如下图。
更多下载安装信息请参考官方文档,在此奉上中文官方文档链接Mmdetection中文官方文档