suvash commited on
Commit
ee64bdc
1 Parent(s): 463babe

gradio article added as well

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. gradio_article.md +19 -4
app.py CHANGED
@@ -22,7 +22,7 @@ with open('gradio_article.md') as f:
22
 
23
  interface_options = {
24
  "title": "Food Image Classifier (Food-101|ResNet50|fast.ai)",
25
- "description": "A food image classifier trained on the Food-101 dataset, using ResNet50 and fast.ai.(https://data.vision.ee.ethz.ch/cvl/datasets_extra/food-101/)",
26
  "article": article,
27
  "examples" : [f'{EXAMPLES_PATH}/{f.name}' for f in EXAMPLES_PATH.iterdir()],
28
  "interpretation": "default",
 
22
 
23
  interface_options = {
24
  "title": "Food Image Classifier (Food-101|ResNet50|fast.ai)",
25
+ "description": "A food image classifier trained on the Food-101 dataset, using ResNet50 via fast.ai.(Dataset from : https://data.vision.ee.ethz.ch/cvl/datasets_extra/food-101/)",
26
  "article": article,
27
  "examples" : [f'{EXAMPLES_PATH}/{f.name}' for f in EXAMPLES_PATH.iterdir()],
28
  "interpretation": "default",
gradio_article.md CHANGED
@@ -1,7 +1,22 @@
1
- # H1
2
 
3
- Some text here
4
 
5
- ## H2
6
 
7
- Some more text here
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ## Dataset
2
 
3
+ The Food-101 dataset, originally made available at https://data.vision.ee.ethz.ch/cvl/datasets_extra/food-101/ is a data set of 101 food categories, with 101'000 images.
4
 
 
5
 
6
+ ## Training
7
+
8
+ Fast.ai was used to train this classifier with a ResNet50 vision learner, without much customization.
9
+
10
+ The final fine tuning of the training loop resulted in the following losses.
11
+
12
+ | epoch | train_loss | valid_loss | error_rate | time |
13
+ |-------|---------------|---------------|---------------|-------|
14
+ | 0 | 1.358123 | 1.100139 | 0.288713 | 05:14 |
15
+ | 1 | 1.129988 | 0.985213 | 0.260693 | 05:12 |
16
+ | 2 | 0.964907 | 0.909715 | 0.241337 | 05:17 |
17
+ | 3 | 0.804738 | 0.843515 | 0.222475 | 05:19 |
18
+ | 4 | 0.638846 | 0.795957 | 0.205347 | 05:16 |
19
+ | 5 | 0.475434 | 0.750069 | 0.192673 | 05:15 |
20
+ | 6 | 0.345060 | 0.742432 | 0.185198 | 05:12 |
21
+ | 7 | 0.247938 | 0.728758 | 0.177624 | 05:12 |
22
+ | 8 | 0.214708 | 0.727486 | 0.177871 | 05:11 |