keras:fit 를 어떻게 사용 합 니까?generator 는 여러 종류의 출력 을 훈련 시 킵 니 다.

1047 단어 keras
이 예 는 매우 간단명료 하 다.모델 은 1 개의 입력,2 개의 출력,두 개의 출력의 가 지 는 각각 MSE 를 손실 로 사용한다.
x = Convolution2D(8, 5, 5, subsample=(1, 1))(image_input)
x = Activation('relu')(x)
x = Flatten()(x)
x = Dense(50, W_regularizer=l2(0.0001))(x)
x = Activation('relu')(x)

output1 = Dense(1, activation='linear', name='output1')(x)
output2 = Dense(1, activation='linear', name='output2')(x)

model = Model(input=image_input, output=[output1, output2])
model.compile(optimizer='adam', loss={'output1': 'mean_squared_error', 'output2': 'mean_squared_error'})

훈련 데 이 터 를 만 드 는 생 성기,여기y=[y1,y2].
batch_generator(x, y, batch_size):
        ....transform images
        ....generate batch batch of size: batch_size 
        yield(X_batch, {'output1': y1, 'output2': y2} ))

이후 호출fit_generator
model.fit_generator(batch_generator(X_train, y_train, batch_size))

질문

좋은 웹페이지 즐겨찾기