matlab-deep-learning/MATLAB-Deep-Learning-Model-Hub

Discover pretrained models for deep learning in MATLAB

解决的问题

此存储库提供了一个集中式中心,用于发现与访问与 MATLAB 兼容的预训练深度学习模型。它通过提供跨多种模态的精选网络列表,消除了用户手动搜索与转换模型的需求,这些模型可用于立即推理或作为迁移学习的起点。

运作方式

此中心作为预训练网络的目录。用户可以通过 MATLAB 函数(例如 imagePretrainedNetwork)或通过链接访问特定的 GitHub 存储库与文档来使用这些模型。模型按任务分类,并提供有关网络大小、准确度以及训练所用特定数据集(例如用于对象检测的 COCO 或用于语义分割的 PASCAL VOC)的详细信息。

目标对象

专为希望将深度学习功能集成到工作流程中,而无需从头开始训练模型的 MATLAB 用户、研究人员与工程师所设计。

重点特色

  • 广泛的模态支持:包含用于计算机视觉(分类、检测、分割、姿态估计)、自然语言处理 (Transformers)、音频(语音转文字、声音分类)、激光雷达(点云分析)与机器人技术(运动规划)的模型。
  • 支持迁移学习:许多模型被特别标注为训练自定义检测器或分类器的起点。
  • 特定任务模型:提供通用模型与特定应用模型,例如车道检测、道路边界检测与脑肿瘤分割。
  • 多样化的架构:包含广泛的架构,包括 YOLO (v2 到 v9)、ResNet、EfficientNet、Vision Transformers 与 GANs。

相关

  • 项目
  • 项目
  • 项目
  • 项目