'How can I show a prediction of trained images in subfolders for such as face mask or no face mask using colab?
Using colab
The image is folder and inside it is subfolders and inside it are images
model.compile(loss='binary_crossentropy',
optimizer=optimizers.RMSprop(learning_rate=2e-4),
metrics=['acc'])
history = model.fit(train_generator,
batch_size=64,
epochs=10)
model.save('model.h5')
Sources
This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.
Source: Stack Overflow
| Solution | Source |
|---|
