OpenMMLab AI实战营笔记-1

OpenMMLab AI实战营笔记-1

OpenMMLab简介

OpenMMLab是香港中文大学-商汤科技联合实验室MMLab开源的算法平台，包含众多SOTA计算机视觉算法。该平台在Github上不是一个单独项目，包括目标检测库MMDetection等库和数据集，涉及超过10个研究方向，开放超过100种算法和600种预训练模型，目前Github总星标超过1.7万。这些库大部分基于深度学习PyTorch框架，算法紧跟前沿，文档丰富，是CV方向系统性较强、社区活跃的开源平台。该平台包括MMEngine、MMCV、MIM、MMClassification、MMDetection、MMDetection3D、MMRotate、MMYOLO、MMSegmentation、MMOCR、MMPose、MMHuman3D、MMSelfSup、MMRazor、MMFewShot、MMAction2、MMTracking、MMFlow、MMEditing、MMGeneration和MMDeploy等项目。

安装教程

OpenMMLab提供了两个版本的MMCV：

• mmcv：功能全面，内置各种CUDA操作，构建时间较长。

• mmcv-lite：轻量版，没有CUDA操作但包含其他所有功能，类似于mmcv<1.0.0。当您不需要这些CUDA操作时，它非常有用。

注意：不要在同一环境中安装两个版本，否则可能会出现错误，例如ModuleNotFound。在可用CUDA的情况下，强烈建议安装完整版本。

先安装Pytorch

如果您需要使用与PyTorch相关的模块，请确保已按照PyTorch官方安装指南在您的环境中成功安装PyTorch。(先装好显卡驱动和CUDA，我默认大家都会装了哈，不赘述。）

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

安装mmcv

在安装mmcv之前，请确保已按照PyTorch官方安装指南成功安装PyTorch。对于Apple Silicon用户，请使用PyTorch 1.13+。

安装mmcv的命令如下：

pip install -U openmim
mim install mmcv

如果您需要指定mmcv的版本，可以使用以下命令：

 mim install mmcv==2.0.0

如果您发现上述安装命令没有使用以.whl结尾的预构建软件包，而是使用以.tar.gz结尾的源码包，则可能没有与PyTorch或CUDA或mmcv版本相对应的预构建软件包，此时您可以从源代码构建mmcv。
有关更多安装方法，请参阅安装文档。

安装mmcv-lite

pip install -U openmim
mim install mmcv-lite

安装完成

(OpenMMLab) PS C:\Users\liu\PycharmProjects\OpenMMLab> python                                                                                      
Python 3.10.11 | packaged by Anaconda, Inc. | (main, Apr 20 2023, 18:56:50) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import mmcv
>>>