Как найти среднее разрешение группы изображений, чтобы я мог подавать размеры в слой CNN в тензорном потоке?

Question

Я работаю над двоичным классификатором изображений в тензорном потоке. Я хочу указать img_shape в слое Conv2D. Я хотел бы знать, есть ли способ найти среднюю форму всех изображений в наборе данных.

Это очень помогло бы. Спасибо

Answer 1

Нашел решение, используя библиотеку PIL (я использую каталог изображений)

import PIL
from PIL import Image

widths = []
heights = []

for img in os.listdir(""):
    img_path = os.path.join("") # Making image file path
    im = Image.open(img_path)
    widths.append(im.size[0])
    heights.append(im.size[1])

AVG_HEIGHT = round(sum(heights)/len(heights))
AVG_WIDTH = round(sum(widths)/len(widths))