PsychicFireSong commited on
Commit
e7f1d35
·
verified ·
1 Parent(s): 9fae6ff

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +29 -6
README.md CHANGED
@@ -1,14 +1,37 @@
1
  ---
2
- title: AMLTestSpace
3
- emoji: 🚀
4
  colorFrom: blue
5
- colorTo: blue
6
  sdk: gradio
7
- sdk_version: 5.49.1
8
  app_file: app.py
9
  pinned: false
10
  license: cc-by-2.0
11
- short_description: For individual assignment, bird classification
12
  ---
13
 
14
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ title: Bird Species Classifier
3
+ emoji: 🐦
4
  colorFrom: blue
5
+ colorTo: green
6
  sdk: gradio
7
+ sdk_version: 4.29.0
8
  app_file: app.py
9
  pinned: false
10
  license: cc-by-2.0
 
11
  ---
12
 
13
+ # COS30082 Bird Species Classifier
14
+
15
+ This project is a deep learning application that classifies bird species from images. It uses a powerful pre-trained model, fine-tuned on the Caltech-UCSD Birds 200 (CUB-200) dataset of 200 different bird species.
16
+
17
+ ## Features
18
+
19
+ - **High-Accuracy Model**: Over 90% accuracy on the CUB-200 dataset test set.
20
+ - **Multiclass Classification**: The model is able to classify images into one of 200 different bird species, with 99% top-5 accuracy.
21
+ - **Interactive Interface**: Live interface built using Gradio allows users to upload images and get predictions quickly.
22
+
23
+ ## Model Details
24
+
25
+ - **Model**: `ConvNeXt V2 Large`
26
+ - **Dataset**: Fine-tuned on a private dataset of 200 bird species.
27
+ - **Frameworks**: PyTorch, timm
28
+ - **Notes**: The model is trained with a convoluted process, using model distillation techniques to produce the final model. The approach is likely very unique, and will require extensive training and tuning to replicate or improve, source code available on request.
29
+
30
+ ## How to Use
31
+
32
+ 1. **Upload an Image**: Click on the "Upload Bird Image" box to select an image file from your computer.
33
+ 2. **Get Predictions**: The model will process the image and display the top 5 most likely bird species, along with their confidence scores.
34
+
35
+ ## License
36
+
37
+ This project is licensed under the cc-by-2.0 License.