代码之家  ›  专栏  ›  技术社区  ›  Prachi

使用MNIST数据集的迁移学习代码中的“filename”是什么?

  •  0
  • Prachi  · 技术社区  · 7 年前

    链接中给出了使用转移学习对MNIST数据集进行分类的代码 https://www.analyticsvidhya.com/blog/2017/06/transfer-learning-the-art-of-fine-tuning-a-pre-trained-model/

    我无法理解代码中的“文件名”是什么?另外,为什么在代码中两次加载数据集。

    我已经看到了使用load_img()函数的代码,但是仍然无法运行给定的代码,因为“filename”是未知的,因为它没有在链接中定义。

    MNIST数据集由“MNIST_train.csv”和“MNIST_test.csv”两个文件组成。有一些代码将.csv文件转换为图像,但是每个图像都有一个.csv文件。在这里,只有两个.csv文件用于列车和测试中显示的所有图像。

    提前谢谢!

    1 回复  |  直到 7 年前
        1
  •  1
  •   Hari Krishnan    7 年前

    数据集采用csv格式,并具有列 filename 其中包含图像名称。 我认为该文件具有以下结构:

       filename    label
    0  file1.jpg    1
    1  file2.jpg    8
    2  file3.jpg    5
    ....
    

    他们把csv文件读入 train

    train=pd.read_csv("R/Data/Train/train.csv")
    

    然后使用循环打开数据帧中给定的每个文件

    for i in range(len(train)):
    temp_img=image.load_img(train_path+train['filename'][i],target_size=(224,224))
    

    使用上述代码加载图像并调整其大小

    train_data = pd.read_csv('train.csv')
    labels = []
    pixels =[]
    for index, row in train_data.iterrows():
        label=np.zeros(10)
        label[row["label"]]=1
        labels.append(label)
        pixels.append(row[1:])
    labels = np.array(labels)
    pixels = np.array(pixels)