我在哪里可以访问 AlexNet?

数据挖掘 神经网络 亚历克斯网
2022-02-26 08:58:47

CNN 上的许多大型图像数据集网站都在谈论从 AlexNet / Imagenet 提供的 1000 个类别中的 120 万张图像的预训练模型开始。这些网站似乎暗示这个数据集是免费提供的,但我实际上无法访问它。例如,我尝试访问https://github.com/deep-diver/AlexNet但无法让 alexnet.ipynb 中的代码运行。考虑以下代码:

for f in data_file.iterdir():
    data = pickle.loads(f.read_bytes(), encoding='bytes')
    if 'meta' in str(f):
        meta = data
    elif 'test' in str(f):
        test = data
    elif 'batch' in str(f.name):
        train_dic = merge(train_dic, data)

这似乎意味着应该有一些数据文件包含文本字符串“meta”、“test”和/或“batch”,但我似乎找不到这些文件。

访问该数据集的最佳位置在哪里——最好是有关于如何下载/使用它的良好文档的地方?

1个回答

ImageNet 数据在线:http ://www.image-net.org/ 。

数据描述说:

14,197,122 张图片,21841 个同义词集索引

但是,您不需要从头开始在所有数据上训练模型,因为您可以使用预先训练的模型,例如在 TF/Keras 中。请参阅文档:https ://keras.io/applications/ 。

我认为原始的 Alexnet 不可用,但许多其他模型已准备好从 Keras 加载。您还可以找到.hdf5在线存储的 Alexnet 版本(您可以使用 Keras/TF 加载)。

我不知道你真正想做什么,但这里有一个简单的例子,说明如何在 Keras 中使用预训练模型(在本例中为 VGG16)。