1 什么是AI模型库

AI模型库是一种集成了各种AI模型和算法的资源库,收集了不同AI任务的预训练SOTA模型,有的甚至提供用户进行模型训练、模型推理和AI应用开发的基础工具和组件。

AI模型库集成了各种经过训练和优化的基础模型和算法,用户可以直接使用所提供的模型进行部署和应用开发,无需从头开始训练,节省了大量的时间和资源。这为开发者提供了更多的时间和精力关注应用的创新和改进,加速了创新的速度。

2 常用的开源AI模型库

2.1 Pytorch Hub

地址:https://pytorch.org/hub/

Pytorch Hub是Pytorch官方提供的预训练AI模型库,提供了很多SOTA预训练模型。

深度学习 – 常用的开源AI模型库-StubbornHuang Blog

2.2 HuggingFace

地址:https://huggingface.co/models

Hugging face 起初是一家总部位于纽约的聊天机器人初创服务商,他们本来打算创业做聊天机器人,然后在github上开源了一个Transformers库,虽然聊天机器人业务没搞起来,但是他们的这个库在机器学习社区迅速大火起来。目前已经共享了超100,000个预训练模型,10,000个数据集,变成了机器学习界的github。

深度学习 – 常用的开源AI模型库-StubbornHuang Blog

2.3 OpenmmLab Model Zoo

地址:https://platform.openmmlab.com/deploee

OpenmmLab是上海人工智能实验室的计算机视觉算法开源体系,OpenmmLab Model Zoo是OpenmmLab提供的硬件模型库,其中包含了640个onnx通用模型和1900+设备专用模型。

深度学习 – 常用的开源AI模型库-StubbornHuang Blog

2.4 ModelScope

地址:https://modelscope.cn/models

ModelScope是阿里达摩院推出的开源模型社区,提供了计算机视觉、自然语言处理、语音、多模态等多个领域的预训练模型。

深度学习 – 常用的开源AI模型库-StubbornHuang Blog

2.5 ModelZoo

地址:https://modelzoo.co/

ModelZoo由新加坡名为Jing Yu Koh本科生牵头搭建,提供不同框架(caffe、pytorch、TensorFlow,JAX等)的不同的任务的开源预训练模型和代码。

深度学习 – 常用的开源AI模型库-StubbornHuang Blog

2.6 Paddle飞桨模型库

热门模型库地址:https://aistudio.baidu.com/modelsoverview

预训练模型库:https://www.paddlepaddle.org.cn/hublist

Paddle飞桨模型库是百度飞桨Paddle提供的AI模型库,覆盖图像、自然语言处理、推荐等多种方向。

深度学习 – 常用的开源AI模型库-StubbornHuang Blog