EnYa32 commited on
Commit
7a35761
Β·
verified Β·
1 Parent(s): c7c8de2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -9
README.md CHANGED
@@ -1,20 +1,31 @@
1
  ---
2
  title: MushroomClassificationCNN
3
- emoji: πŸš€
4
  colorFrom: red
5
  colorTo: red
6
- sdk: docker
7
- app_port: 8501
8
- tags:
9
- - streamlit
10
  pinned: false
11
  short_description: Mushroom image classification with ResNet50
12
  license: mit
13
  ---
14
 
15
- # Welcome to Streamlit!
16
 
17
- Edit `/src/streamlit_app.py` to customize this app to your heart's desire. :heart:
18
 
19
- If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
20
- forums](https://discuss.streamlit.io).
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  title: MushroomClassificationCNN
3
+ emoji: πŸ„
4
  colorFrom: red
5
  colorTo: red
6
+ sdk: streamlit
7
+ app_file: src/streamlit_app.py
 
 
8
  pinned: false
9
  short_description: Mushroom image classification with ResNet50
10
  license: mit
11
  ---
12
 
13
+ # πŸ„ Mushroom Classifier (ResNet50)
14
 
15
+ This app classifies mushroom images into **10 classes** using **Transfer Learning with ResNet50**.
16
 
17
+ ## βœ… Model
18
+ - Backbone: **ResNet50** (ImageNet pretrained)
19
+ - Input: **224 Γ— 224 RGB**
20
+ - Preprocessing: `tf.keras.applications.resnet50.preprocess_input`
21
+ - Metrics used during training: **accuracy** and **top-3 accuracy**
22
+
23
+ ## πŸ“Œ Results (Validation)
24
+ - **Accuracy:** 0.8520
25
+ - **Top-3 Accuracy:** 0.9704
26
+ - **Loss:** 0.6097
27
+
28
+ ## πŸš€ How to run locally
29
+ ```bash
30
+ pip install -r requirements.txt
31
+ streamlit run app.py