Вопрос о размере кераса - PullRequest
       4

Вопрос о размере кераса

0 голосов
/ 07 октября 2019

Я использую keras ImageDataGenerator и flow_from_directory для проблемы регрессии. Есть некоторые параметры, которые я не могу выяснить.

Вот моя структура данных:

 data\
        1st label folder(e.g. 1.3)\
            image for 1st label
            ...
        2nd label folder\
            image for 2nd label
            ...
        ...

и вотнекоторые коды:

gen = ImageDataGenerator(width_shift_range=0.2,height_shift_range=0.1,..........)
train_datagen = gen.flow_from_directory(path, batch_size=?, class_mode='sparse')

def regression_flow_from_directory(gen, label_list):
    for x, y in flow_from_directory_gen:
        y=y.astype('int32')
        yield x, label_list[y]

model.fit_generator(train_datagen ,steps_per_epoch=?,.......)

Q1. Если в каждой папке ярлыков есть несколько изображений (скажем, 2). Какой номер плеча я установил для batch_size?

Q2. Относится к Q1. Если у меня есть 10000 меток папки (т.е. 20000 изображений в общей сложности). Какой номер плеча я установил для steps_per_epoch?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...