由于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]