Beans VGG16
What it is
A three-class bean-leaf disease classifier built with tf.keras.applications.VGG16(include_top=False) and a global-average-pooling, dense, dropout and softmax classification head. The VGG weights are licensed under CC BY 4.0.
Training
Transfer learning: ImageNet weights ported by Keras Applications, base frozen, new head trained, no fine-tuning.
Dataset and split
Beans with 1,034 training, 133 validation and 128 test images. Dataset source credit: Makerere AI Lab; source and licence details are to be verified.
Preprocessing and input
Input is 224×224 RGB. tf.keras.applications.vgg16.preprocess_input is embedded in the saved model.
Result
Test accuracy: 84.38% — from saved notebook output, not re-run.
Loading
import keras
model = keras.saving.load_model("beans_vgg16.keras")
Limitations
This is coursework and was not re-run offline for this model card. The Beans test split is small at 128 images, so the reported result has limited statistical coverage. This model is not for production use.
Repository: https://github.com/faiz-sheikh/Deep-Learning
- Downloads last month
- 14