张量流中预训练的 alexnet

发布于 2025-01-11 17:40:16 字数 1539 浏览 0 评论 0原文

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。

评论(1

榆西 2025-01-18 17:40:16

截至目前,Keras(官方)似乎没有提供预训练的 AlexNet 模型。另一方面,PyTorch 可以。如果您愿意使用不同的框架来完成任务,则可以使用 PyTorch。您可以像这样检索 AlexNet 的预训练版本:

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

您可以找到可用预训练模型的列表 此处,以及图像分类的迁移学习教程 这里

希望能回答您的问题!

As of right now, Keras does not (officially) seem to offer a pre-trained AlexNet model. PyTorch, on the other hand, does. If you are willing to use a different framework for the task, you can use PyTorch. You can retrieve a pre-trained version of the AlexNet like so:

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

You can find the list of available pre-trained models here, and a transfer learning tutorial for image classification here.

Hope that answers your question!

~没有更多了~
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文