如何使用Python从头开始训练AlexNet？

Question

我应该从头开始训练一个亚历克斯内特，因为我得到的那个不够好（实际上一点也不好）。你能推荐一些脚本吗？也许是用Python编写的。我想有些东西已经在起作用了，我不想重新发明轮子。我已经有了我的图像集来训练网络。在

Note:- This is not my "homework". I'm expected to develop a Java application that uses such a network

谢谢！在

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)