Я сохраняю матрицу изображения в списке X и их номер класса в списке Y примерно для 30000 изображений в наборе данных Caltech 256.
from PIL import Image
base_dir = os.path.join('D:\\256_ObjectCategories')
for f in sorted(os.listdir(base_dir)):
if os.path.isdir(base_dir+ '\\' +f):
print(f"{f} is a target class")
for i in sorted(os.listdir(base_dir+ '\\' +f)):
#print(f"{i} is an input image path")
im_1 = Image.open(base_dir+ '\\' +f + '\\' + i)
ar = np.array(im_1)
X.append(ar)
#Y.append(f)
Y.append(int(f[0:3])-1)#saves the number 249 for the folder '250.zebra'
Это занимает очень много времени. Я хотел бы знать, как вызвать GPU для этой операции.