有没有一种方法可以有效地从fastai.data.core.Dataset对象中提取所有PILLImages



由于GPU不兼容,我试图将代码从fastai1重写为fastai2,但在提取图像时遇到了问题:

fastai1(从train_ds中提取所有图像(:

data.train_ds.x

我有100000个图像作为输入,但我的列表理解技术不起作用(每次我试图运行它时,线程都会被杀死(:

[x[0] for x in data.train_ds]

有更好的方法提取所有图像吗?

[x[0] for x in data.train_ds]

同时将所有图像加载到内存中,解决此问题的正确方法是在数据集创建中加载它们。

这个问题的解决方案各不相同:由于我需要这些图像来创建自定义的fastai数据集,我决定只在数据集中加载图像,并将data.train_ds传递到数据集。

代码是

CustomDataset(data.train_ds,...)

在自定义数据集中(可能在getitem((中(使用:

self.images[i][0] 

最新更新