python - 如何使用 Python 从头开始​​训练 Alex-Net？

Question

我应该从头开始训练一个 AlexNet，因为给我的那个不够好（实际上它一点都不好）。你能推荐一些脚本吗，也许在 Python 中可以完成这项工作。我想有些东西已经在起作用了，我不想重新发明轮子。我已经有了我的图像集来训练网络。

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

谢谢 ！

标签： pythonconv-neural-network

Answer 1

你可以使用PyTorch。这个 Python 库提供了一堆预训练的网络。

通常对于alexnet，一旦安装了 pytorch，您就可以像这样继续：

import torch
import torchvision.models.alexnet

model = alexnet(pretrained=False)