tracel-ai/models

解决的问题

此存储库提供了一个使用 Burn 深度学习框架构建的深度学习模型实现与示例的集中式集合。它让开发者能够轻松找到并整合跨不同模态的各种现有模型架构。

运作方式

该项目作为官方实现与社区贡献模型的中心。它根据功能（如语言模型、图像分类与语音识别）对模型进行分类，并提供代码托管的特定存储库链接。

目标对象

使用 Burn 框架且需要在项目中实现现成模型架构的开发者与研究人员。

特色亮点

• 多元模态：包含文本 (ALBERT, RoBERTa, Llama 2, RWKV v7)、图像 (ResNet, MobileNetV2, YOLOX, Stable Diffusion) 与音频/语音 (Whisper, Wav2Vec 2.0, OmniVoice) 模型。
• 官方与社区支持：结合了一系列官方 Burn 模型与精选的社区开发实现。
• 广泛的应用范围：涵盖从对象检测与 OCR 到语音克隆与乐谱检测的所有领域。