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

时间:2019-06-20 11:43:26

标签: python conv-neural-network

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

  

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

谢谢!

2 个答案:

答案 0 :(得分:0)

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

答案 1 :(得分:0)

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

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

import torch
import torchvision.models.alexnet

model = alexnet(pretrained=False)