AtharAbbas993's picture
Update README.md
6a6d0e8 verified

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:

  1. daisy ๐ŸŒผ - Classic white petals with yellow center
  2. dandelion ๐ŸŒž - Bright yellow composite flowers
  3. rose ๐ŸŒน - Layered petals in various colors
  4. sunflower ๐ŸŒป - Large yellow flowers with dark centers
  5. 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