| license: apache-2.0 | |
| tags: | |
| - food | |
| - classification | |
| - efficientnet | |
| - nutrition5k | |
| pipeline_tag: image-classification | |
| # EfficientNet-B2 Food Classifier | |
| Classifies food crops into 37 categories from the Nutrition5k dataset. | |
| Used as Stage 2 in a food segmentation + identification pipeline. | |
| **Stage 1 segmenter:** [arunapb/yolo11l-food-segmentation](https://huggingface.co/arunapb/yolo11l-food-segmentation) | |
| ## Classes (37) | |
| 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 | |
| ## Performance | |
| Best validation accuracy: 0.9875 | |
| ## Usage | |
| See class_names.json for the full class list. | |
| Load best_classifier.pth with torchvision EfficientNet-B2. | |