1、Tensorflow官网模型下载:

https://github.com/tensorflow/models/tree/master/research/slim#Pretrained

界面显示:

2、pytorch官网:

点击直接下载即可

model_urls = {
    'resnet18': 'https://download.pytorch.org/models/resnet18-5c106cde.pth',
    'resnet34': 'https://download.pytorch.org/models/resnet34-333f7ec4.pth',
    'resnet50': 'https://download.pytorch.org/models/resnet50-19c8e357.pth',
    'resnet101': 'https://download.pytorch.org/models/resnet101-5d3b4d8f.pth',
    'resnet152': 'https://download.pytorch.org/models/resnet152-b121ed2d.pth',
}

3、不同深度学习框架的的模型转换:

https://github.com/ysh329/deep-learning-model-convertor

里面有很多常用的模型转换,如caffe、tensorflow、pytorch、mxnet等

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐