OpenMMLab 训练营第二期 笔记一

OpenMMLab是一个基于PyTorch的开源平台,涵盖了多个视觉库和算法,包括预训练模型。其项目涉及图像分类、预训练、语义分割、目标检测、姿态估计等多个领域,提供丰富的工具箱和测试基准,如MMPreTrain、MMSegmentation、MMDetection和MMPose。此外,它还支持AI内容生成和模型部署。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

OpenMMLab

OpenMMLab 基于 PyTorch,涵盖 30+ 个视觉库,实现了 300+ 种算法,并包含 2400+ 个预训练模型。
OpenMMLab框架

OpenMMLab 包括以下项目

  • MMEngine: OpenMMLab 深度学习模型训练基础库
  • MMCV: OpenMMLab 计算机视觉基础库
  • MIM: MIM 是 OpenMMlab 项目、算法、模型的统一入口
  • MMEval: 统一开放的跨框架算法评测库
  • MMPreTrain: OpenMMLab 深度学习预训练工具箱
  • MMDetection: OpenMMLab 目标检测工具箱
  • MMDetection3D: OpenMMLab 新一代通用 3D 目标检测平台
  • MMRotate: OpenMMLab 旋转框检测工具箱与测试基准
  • MMYOLO: OpenMMLab YOLO 系列工具箱与测试基准
  • MMSegmentation: OpenMMLab 语义分割工具箱
  • MMOCR: OpenMMLab 全流程文字检测识别理解工具包
  • MMPose: OpenMMLab 姿态估计工具箱
  • MMHuman3D: OpenMMLab 人体参数化模型工具箱与测试基准
  • MMSelfSup: OpenMMLab 自监督学习工具箱与测试基准
  • MMRazor: OpenMMLab 模型压缩工具箱与测试基准
  • MMFewShot: OpenMMLab 少样本学习工具箱与测试基准
  • MMAction2: OpenMMLab 新一代视频理解工具箱
  • MMTracking: OpenMMLab 一体化视频目标感知平台
  • MMFlow: OpenMMLab 光流估计工具箱与测试基准
  • MMagic: OpenMMLab 新一代人工智能内容生成(AIGC)工具箱
  • MMGeneration: OpenMMLab 图片视频生成模型工具箱
  • MMDeploy: OpenMMLab 模型部署框架
  • Playground: 收集和展示 OpenMMLab 相关的前沿、有趣的社区项目

下面是本次训练营课程涉及的 5 个项目

MMPreTrain

MMPreTrain 是做图像分类、预训练和多模态的算法库。
提供了77个算法,508个预训练模型。

支持多种开箱即用的推理任务

  • 图像分类 (Image Classification)
  • 图像描述 (Image Caption)
  • 视觉问答 (Visual Question Answering)
  • 视觉定位 (Visual Grounding)
  • 检索 (Image-To-Image, Text-To-Image, Image-To-Text)

MMSegmentation

MMSegmentation 是专门做图像分割,特别是语义分割的算法库。在街景、无人驾驶、遥感图像和医疗领域等领域语义分割都是非常重要的算法。语义分割最常用的场景就是在医疗影像分析和卫星遥感分析。

MMSegmentation 提供了600多个预训练模型,复现了40多篇论文。同时,也可以用他来训练自己的语义分割算法。

MMDetection

其中最知名算法库就是 MMDetection 目标检测算法库,在业界有很大影响力,在 github 上有 24k star。

MMDetection 可以解决目标检测,实例分割,全景分割和目标追踪任务,不仅实现了这些领域上的经典算法,还包括前沿顶会论文的复现,例如 Faster R-CNN,RetinaNet,SSD,YOLOv3,RTMDet,DiffusionDet 等。

MMPose

MMPose 是专门做姿态估计的算法库。姿态估计其实就是关键点检测,例如把一个人的关节骨架的点识别出来,然后串成一个火柴人。
MMPose支持人脸,人体手动物和服装的关键点检测,既有2d又有3d。

MMagic

MMagic 从这个名字我们就能看出来他是一个魔法库 😃。他是人工智能内容生成(AIGC)工具箱。

参考资料

[^1]OpenMMLabCamp AI 实战营
[^2]Github八万star,开源计算机视觉神器OpenMMLab

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值