Блог

Я пытаюсь обработать изображения для gan, над которым работаю, но по какой-то причине я не могу этого сделать и не могу сказать почему. Любая помощь будет признательна.

 IMAGE_SIZE = 128
IMAGE_CHANNELS = 3
IMAGE_DIR = 'dataset/'
images_path = '/content/drive/MyDrive/Art/Abstract'
training_data = []
print('resizing...')

for filename in tdqm(os.listdir(images_path)):
  path = os.path.join(images_path, filename)
  image = Image.open(path).resize((IMAGE_SIZE, IMAGE_SIZE), Image.ANTIALIAS)

  training_data.append(np.asarray(image))

training_data = np.reshape(training_data, (-1, IMAGE_SIZE, IMAGE_SIZE, IMAGE_CHANNELS))
training_data = training_data / 127.5 - 1

print('saving file...')
np.save('/content/drive/MyDrive/Art/cubism_data.npy', training_data)