欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

特征提取网络(分类)

程序员文章站 2022-07-14 15:19:35
...

以下网络可通过torchvision.models导入

import torchvision.models as models

resnet18 = models.resnet18(pretrained=True)
vgg16 = models.vgg16(pretrained=True)
alexnet = models.alexnet(pretrained=True)
squeezenet = models.squeezenet1_0(pretrained=True)

1. AlexNet

特征提取网络(分类)

2. vgg

特征提取网络(分类)

3. googlenet

特征提取网络(分类)
特征提取网络(分类)
https://my.oschina.net/u/876354/blog/1637819

4. resnet

特征提取网络(分类)
特征提取网络(分类)

5. squeezenet

相关标签: # 图像分类