在tensorflow中预先训练的alexnet

Question

我想使用预培训的Alexnet进行转移学习。我在Keras库里找不到它。我是不是漏掉了什么？我在这里看到的另一种选择是创建模型

1. 载荷预训练权重
2. 从无到有

由于资源限制，无法从头开始使用imagenet dataset进行培训。加载预先训练的重量将工作。你能为Alexnet提供获得预训练体重的指点吗？

谢谢,

Answer 1

到目前为止，Keras似乎没有(正式地)提供经过培训的AlexNet模型。另一方面，PyTorch确实如此。如果您愿意为该任务使用不同的框架，则可以使用PyTorch。您可以检索经过预处理的AlexNet版本，如下所示：

import torchvision.models as models
alexnet = models.alexnet(pretrained=True)

您可以找到可用的预培训模型这里的列表，以及图像分类这里的转移学习教程。

希望这能回答你的问题！