如何使用Python从头训练Alex-Net？

Question

我应该从头开始训练AlexNet，因为给我的那个不够好（实际上一点也不好）。您能推荐一些脚本吗，也许是用Python来完成这项工作。我想已经有一些工作了，我不想重新发明轮子。我已经有了 imageset 来训练网络。

  

注意：-这不是我的“作业”。我期望开发使用这种网络的Java应用程序

谢谢！

Answer 1

只需几行代码，您就可以使用Keras在python中进行操作。 看看这个例子https://engmrk.com/alexnet-implementation-using-keras/

Answer 2

您可以使用PyTorch。这个Python库提供了许多预训练的网络。

对于alexnet，通常，安装pytorch后，您可以像这样进行操作：

import torch
import torchvision.models.alexnet

model = alexnet(pretrained=False)