Получение матрицы изображения с помощью Pillow занимает слишком много времени. ? Как я могу использовать графический процессор для этой операции? Какие еще пакеты займут меньше времени? - PullRequest
0 голосов
/ 06 мая 2020

Я сохраняю матрицу изображения в списке 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 для этой операции.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...