Skip to content

Instantly share code, notes, and snippets.

@Yuvnish017

Yuvnish017/training_model.py

Created June 29, 2021 06:27
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Yuvnish017/94284b1a4d31da07f33102da08c53dcd to your computer and use it in GitHub Desktop.
Save Yuvnish017/94284b1a4d31da07f33102da08c53dcd to your computer and use it in GitHub Desktop.
Download ZIP
Driver Drowsiness Detection
Raw
training_model.py
aug = ImageDataGenerator(rotation_range=20,
zoom_range=0.2,
horizontal_flip=True)
hist = model.fit(aug.flow(X_train, Y_train, batch_size=128),
batch_size=128,
epochs=200,
validation_data=(X_test, Y_test))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment