arunapb commited on
Commit
c0b2158
·
verified ·
1 Parent(s): d95b7f7

Add model card

Browse files
Files changed (1) hide show
  1. README.md +26 -0
README.md ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - food
5
+ - classification
6
+ - efficientnet
7
+ - nutrition5k
8
+ pipeline_tag: image-classification
9
+ ---
10
+
11
+ # EfficientNet-B2 Food Classifier
12
+
13
+ Classifies food crops into 37 categories from the Nutrition5k dataset.
14
+ Used as Stage 2 in a food segmentation + identification pipeline.
15
+
16
+ **Stage 1 segmenter:** [arunapb/yolo11l-food-segmentation](https://huggingface.co/arunapb/yolo11l-food-segmentation)
17
+
18
+ ## Classes (37)
19
+ almond, apple, asparagus, avocado, broccoli, carrot, cauliflower, celery_stick, cherry, chicken_duck, corn, cucumber, dried_cranberries, egg, eggplant, enoki_mushroom, grape, green_beans, lettuce, melon, olives, onion, orange, pasta, pear, pepper, pineapple, pizza, potato, rice, salad, sausage, steak, tofu, tomato, watermelon, wonton_dumplings
20
+
21
+ ## Performance
22
+ Best validation accuracy: 0.9875
23
+
24
+ ## Usage
25
+ See class_names.json for the full class list.
26
+ Load best_classifier.pth with torchvision EfficientNet-B2.