Plant Species Classification Model
Model Description
This is a deep learning model for automated classification of flower species using computer vision. The model can identify 5 common flower types with high accuracy.
Model type: Image Classification
Architecture: EfficientNetB0 with custom classification head
Input: 224ร224 RGB images
Output: 5-class classification probabilities
Intended Uses
Primary Use Cases
- ๐ฟ Educational plant identification tools
- ๐ฑ Mobile flower recognition applications
- ๐ฌ Botanical research and biodiversity monitoring
- ๐ธ Gardening and nature enthusiast apps
Limitations
- Trained on only 5 specific flower species
- Performance may vary with image quality and lighting conditions
- Not suitable for rare or unusual flower varieties
Classes
The model classifies images into 5 flower species:
- daisy ๐ผ - Classic white petals with yellow center
- dandelion ๐ - Bright yellow composite flowers
- rose ๐น - Layered petals in various colors
- sunflower ๐ป - Large yellow flowers with dark centers
- tulip ๐ท - Cup-shaped flowers in vibrant colors
Training Data
- Dataset: Flowers Recognition from Kaggle
- Total Images: ~4,300
- Split: 80% training, 20% validation
- Augmentation: Rotation, flipping, zooming, brightness adjustment
Performance
- Validation Accuracy: >90%
- Inference Speed: Real-time capable
- Model Size: ~30MB
Usage
from tensorflow.keras.models import load_model
from tensorflow.keras.preprocessing import image
import numpy as np
# Load and use the model for flower classification
model = load_model('flower_classification_model.h5')
Input Requirements:
- Image format: JPEG, PNG
- Image size: 224ร224 pixels
- Color mode: RGB
Ethical Considerations
- Intended for educational and research purposes
- Should not replace expert botanical identification
- Respect privacy when deploying in applications
Citation
If you use this model in your work, please cite:
Plant Species Classification Model by Athar Abbas
https://huggingface.co/AtharAbbas993/Plant_Species_Classification