다음은 데이터셋 전처리를 실행한 결과입니다.
Found 385 images belonging to 2 classes.
Found 98 images belonging to 2 classes.
이제 텐서보드를 설정하고 모델 훈련을 시작합니다.
코드 6-12 텐서보드 설정 및 모델 훈련
log_dir = "../chap6/img/log6-2/"
tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir=log_dir,
histogram_freq=1, profile_batch=0)
model.fit(train_generator,
epochs=EPOCHS,
steps_per_epoch=train_num // BATCH_SIZE,
validation_data=valid_generator,
validation_steps=valid_num // BATCH_SIZE,
callbacks=[tensorboard_callback],
verbose=1)