使用Python枕头在Google Colab上调整图像大小

Question

我的ipynb和一个名为PRimage的文件夹（带有100多个图像）在我的Google驱动器中，并且我的驱动器已经安装在/ content / drive中。我的图像按顺序排列。 1_1.jpg，1_2.jpg等。我正在尝试使用for循环来调整所有图像的大小，例如：

from google.colab import drive
drive.mount('/content/drive')

from os import listdir
from matplotlib import image
from PIL import Image

loaded_images = list()
for filename in listdir('/content/drive/My Drive/PRimage'):
  img_data = image.imread('/content/drive/My Drive/PRimage/'+ filename)
  loaded_images.append(img_data)
  print('> loaded %s %s' % (filename, img_data.shape))

def resize():
    files = listdir('/content/drive/My Drive/PRimage')
    for item in files:
            image = Image.open(item)
            image.thumbnail((64,64))
            print(image.size)

resize()

但是，我收到此错误消息：

Answer 1

插入新的代码单元并使用pwd检查您的当前工作目录。确保它位于/content/drive/My Drive/PRimage。用cd /content/drive/My\ Drive/PRimage更改目录。您的FileNotFoundError是未知密码的原因。在这种情况下，请始终寻找您的根本工作方向。您的代码从pwd执行，并期望其中包含类似的目录结构。

帮助功能调整图像大小

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)