Downloads · 30 days
2
11% of all-time downloads
AlphaAkib/food-nutrition-multitask
food-nutrition-multitask is a machine learning model from AlphaAkib. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
ConvNeXt-Tiny backbone fine-tuned jointly for: - Nutrition regression: totalcalories, totalmass, totalfat, totalcarb, totalprotein - Multi-label ingredient recognition (80 ingredient classes)
Downloads · 30 days
2
11% of all-time downloads
All-time downloads
19
Public
Repo size
113 MB
Likes
0
Public
Click a slice to open those files.
.pt113 MB · 100%
From the Hugging Face model README
ConvNeXt-Tiny backbone fine-tuned jointly for:
Trained on mmathys/food-nutrients (Nutrition5k-derived).
Test-set MAE: {'total_calories': 39.68, 'total_mass': 23.8, 'total_fat': 2.81, 'total_carb': 3.56, 'total_protein': 3.51} Test-set ingredient micro-F1: 0.881
See the accompanying notebook for the model definition (FoodNutritionModel) and how to load
model_weights.pt + config.json for inference.