adds model card
Browse files
README.md
CHANGED
|
@@ -12,4 +12,62 @@ tags:
|
|
| 12 |
- biology
|
| 13 |
- image-classification
|
| 14 |
- vision
|
| 15 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
- biology
|
| 13 |
- image-classification
|
| 14 |
- vision
|
| 15 |
+
---
|
| 16 |
+
|
| 17 |
+
# Model Card for "Bird Species Classifier"
|
| 18 |
+
|
| 19 |
+
## Model Description
|
| 20 |
+
The "Bird Species Classifier" is a state-of-the-art image classification model designed to identify various bird species from images. It uses the EfficientNet architecture and has been fine-tuned to achieve high accuracy in recognizing a wide range of bird species.
|
| 21 |
+
|
| 22 |
+
### How to Use
|
| 23 |
+
You can easily use the model in your Python environment with the following code:
|
| 24 |
+
|
| 25 |
+
```python
|
| 26 |
+
from transformers import AutoFeatureExtractor, AutoModelForImageClassification
|
| 27 |
+
|
| 28 |
+
extractor = AutoFeatureExtractor.from_pretrained("chriamue/bird-species-classifier")
|
| 29 |
+
model = AutoModelForImageClassification.from_pretrained("chriamue/bird-species-classifier")
|
| 30 |
+
```
|
| 31 |
+
|
| 32 |
+
### Applications
|
| 33 |
+
- Bird species identification for educational or ecological research.
|
| 34 |
+
- Assistance in biodiversity monitoring and conservation efforts.
|
| 35 |
+
- Enhancing user experience in nature apps and platforms.
|
| 36 |
+
|
| 37 |
+
## Training Data
|
| 38 |
+
The model was trained on the "Bird Species" dataset, which is a comprehensive collection of bird images. Key features of this dataset include:
|
| 39 |
+
|
| 40 |
+
- **Total Species**: 525 bird species.
|
| 41 |
+
- **Training Images**: 84,635 images.
|
| 42 |
+
- **Validation Images**: 2,625 images.
|
| 43 |
+
- **Test Images**: 2,625 images.
|
| 44 |
+
- **Image Format**: Color images (224x224x3) in JPG format.
|
| 45 |
+
- **Source**: Sourced from Kaggle.
|
| 46 |
+
|
| 47 |
+
## Training Results
|
| 48 |
+
The model achieved impressive results after 6 epochs of training:
|
| 49 |
+
|
| 50 |
+
- **Accuracy**: 96.8%
|
| 51 |
+
- **Loss**: 0.1379
|
| 52 |
+
- **Runtime**: 136.81 seconds
|
| 53 |
+
- **Samples per Second**: 19.188
|
| 54 |
+
- **Steps per Second**: 1.206
|
| 55 |
+
- **Total Training Steps**: 31,740
|
| 56 |
+
|
| 57 |
+
These metrics indicate a high level of performance, making the model reliable for practical applications.
|
| 58 |
+
|
| 59 |
+
## Limitations and Bias
|
| 60 |
+
- The performance of the model might vary under different lighting conditions or image qualities.
|
| 61 |
+
- The model's accuracy is dependent on the diversity and representation in the training dataset. It may perform less effectively on bird species not well represented in the dataset.
|
| 62 |
+
|
| 63 |
+
## Ethical Considerations
|
| 64 |
+
This model should be used responsibly, considering privacy and environmental impacts. It should not be used for harmful purposes such as targeting endangered species or violating wildlife protection laws.
|
| 65 |
+
|
| 66 |
+
## Acknowledgements
|
| 67 |
+
We would like to acknowledge the creators of the dataset on Kaggle for providing a rich source of data that made this model possible.
|
| 68 |
+
|
| 69 |
+
## See also
|
| 70 |
+
|
| 71 |
+
- [Bird Species Dataset](https://huggingface.co/datasets/chriamue/bird-species-dataset)
|
| 72 |
+
- [Kaggle Dataset](https://www.kaggle.com/datasets/gpiosenka/100-bird-species/data)
|
| 73 |
+
- [Bird Species Classifier](https://huggingface.co/dennisjooo/Birds-Classifier-EfficientNetB2)
|