如何将 parquet 文件中的图像数据和 csv 文件中的标签加载到 fastai答案

【问题标题】：How to load Image data from parquet file and labels from csv file to fastai如何将 parquet 文件中的图像数据和 csv 文件中的标签加载到 fastai
【发布时间】：2019-12-22 19:47:54
【问题描述】：

我正在尝试在https://www.kaggle.com/c/bengaliai-cv19 中使用 fastai 库，但找不到任何有关从 parquet 文件加载图像的文档。

我已将 parquet 文件转换为 dataFrame 并且 ImageDataBunch 需要参数 path 但图像已经在 dataFrame 中，如何将其加载到 ImageDataBunch 中。

【问题讨论】：

标签： python-3.x machine-learning fast-ai

【解决方案1】：

您可以遍历数据框并将图像保存到文件夹中，然后使用ImageDataBunch 类的from_folder 工厂函数。

从 kaggle 网站的数据描述来看，parquet 文件的每一行似乎都是形状为137 x 236 的测试图像。

  image_id    0    1    2    3    4    5    6    7    8  .....
0   Test_0  247  253  253  252  252  252  252  253  253  .....
1   Test_1  253  253  253  253  253  253  253  253  253  .....
2   Test_2  253  253  253  253  253  252  251  252  252  .....
.
.
.
.

我已经试过这个来获得第一张图片。

# get the image
>>> img = df.iloc[1,1:].to_numpy().astype(int).reshape(137,236)
>>> img.shape
(137, 236)

# Image_id
>>> df.iloc[1,0]
'Test_1'

希望有帮助！

【讨论】：

这是否意味着fastai API仅适用于图像路径而不适用于图像数据，我可以将图像保存在kaggle上。