Как обрезать изображения разной длины и ширины до равных длины и ширины? - PullRequest
Новая
       42

Как обрезать изображения разной длины и ширины до равных длины и ширины?

0 голосов
/ 28 мая 2019

Я использую набор данных COCO. в этом наборе данных изображения имеют разную длину и ширину, но я должен обрезать их до равной длины и ширины.

Все изображения размещены в папке. Я попытался обрезать их до равной длины и ширины, то есть 386X386 

import sys
import os
from PIL import Image

filepath = "/content/drive/My Drive/Colab Notebooks/SRGAN/data"

# Loop through all provided arguments
for filename in os.listdir(filepath):
    if "." not in filename:
        continue
    ending = filename.split(".")[1]
    if ending not in ["jpg", "gif", "png"]:
        continue


    # Attempt to open an image file
    image = Image.open(os.path.join(filepath, filename))


    # Perform operations on the image here
    image = image.crop((0, 0, 386, 386))

    # Split our origional filename into name and extension 
    name, extension = os.path.splitext(filename)

    # Save the image as "(origional_name)_thumb.jpg
    print(name + '_cropped.jpg')
    image.save(os.path.join("/content/drive/My Drive/Colab Notebooks/SRGAN/data", name + '_cropped.jpg'))

Он должен обрезать все изображения до размера 386X386, но генерирует несколько копий одних и тех же изображений разной длины и ширины.

1 Ответ

0 голосов
/ 28 мая 2019 
from keras.preprocessing import image
test_img = image.load_img('Image Path', target_size=(386,386))

«целевой размер» определяет размеры для обрезки

...