深度学习-分享几个开源AI模型库

现在市面上有很多的深度学习预训练模型库，这些预训练库收集了不同AI任务的SOTA模型，有的还提供推理的SDK。这对于做AI应用和AI开发来说，是宝贵的开源资源。

今天分享几个AI预训练模型库，希望对你有帮助。

• pytorch hub

  pytorch Hub 提供了许多SOTA Pre-trained模型

  • 经典的分类模型 Resnet50，Inception_v3，ResNext，Densenet，EfficientNet等
  • 目标检测模型 SSD，YOLOP，YOLOv5
  • PyTorch-Transformers
  • NLP：RoBERTa

  使用可以通过load方式加载模型

  import torch
  model = torch.hub.load('pytorch/vision', 'resnet18', pretrained=True)

  

• ModelZoo

  ModelZoo（https://modelzoo.co/）提供不同框架（caffe、pytorch、TensorFlow，JAX等）的不同的任务的开源预训练模型和代码github

  

• modelscope
  modelscope（魔搭社区）是阿里达摩院推出的开源模型社区，不仅有预训练模型，还有推理SDK和GPU环境。

  

• Hugging Face

  Hugging Face最早是以提供Transformer预训练模型起家的，是一个优秀的开源社区。

  • https://huggingface.co/models