Вставьте новую ячейку кода и используйте pwd
, чтобы проверить текущий рабочий каталог . Убедитесь, что это на /content/drive/My Drive/PRimage
. Используйте cd /content/drive/My\ Drive/PRimage
, чтобы изменить каталоги. Ваш FileNotFoundError
является причиной вашего неизвестного pwd. Всегда ищите ваши root рабочие каталоги в таких случаях. Ваш код выполняется из pwd и ожидает аналогичную структуру dir внутри него.
Вспомогательная функция для изменения размера изображения
def resize_image(src_img, size=(64,64), bg_color="white"):
from PIL import Image
# rescale the image so the longest edge is the right size
src_img.thumbnail(size, Image.ANTIALIAS)
# Create a new image of the right shape
new_image = Image.new("RGB", size, bg_color)
# Paste the rescaled image onto the new centered background
new_image.paste(src_img, (int((size[0] - src_img.size[0]) / 2), int((size[1] - src_img.size[1]) / 2)))
# return the resized image
return new_image
# get the list of test image files
test_folder = '/content/drive/My Drive/PRimage'
test_image_files = os.listdir(test_folder)
# Empty array on which to store the images
image_arrays = []
size = (64,64)
background_color="white"
# Get the images
for file_idx in range(len(test_image_files)):
img = Image.open(os.path.join(test_folder, test_image_files[file_idx]))
# resize the image
resized_img = np.array(resize_image(img, size, background_color))
# Add the image to the array of images
image_arrays.append(resized_img)