特征提取网络(分类)
程序员文章站
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