Spaces:
Running
Running
gradio article added as well
Browse files- app.py +1 -1
- 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
|
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 |
-
|
2 |
|
3 |
-
|
4 |
|
5 |
-
## H2
|
6 |
|
7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
|