Hugging Face's logo

Kaludi
/
food-category-classification

Image Classification
Transformers
PyTorch
swin
vision
Inference Endpoints
Model card Files Community
1
food-category-classification / README.md
Kaludi's picture
Kaludi
Update README.md
03fa9bd
preview code
|
raw
history blame contribute delete
2.71 kB
metadata
tags:
  - vision
  - image-classification
datasets:
  - Kaludi/data-food-classification
widget:
  - src: >-
      https://static01.nyt.com/images/2021/08/30/dining/as-grilled-cheese-sandwich-on-the-grill/as-grilled-cheese-sandwich-on-the-grill-articleLarge.jpg
    example_title: Bread
  - src: https://cdn.britannica.com/34/176234-050-0E0C55C6/Glass-milk.jpg
    example_title: Dairy Product
  - src: >-
      https://images-gmi-pmc.edge-generalmills.com/7c1096c7-bfd0-4806-a794-1d3001fe0063.jpg
    example_title: Dessert
  - src: >-
      https://theheirloompantry.co/wp-content/uploads/2022/06/how-to-fry-eggs-perfectly-in-4-ways-the-heirloom-pantry.jpg
    example_title: Egg
  - src: >-
      https://www.mashed.com/img/gallery/the-real-reason-fried-foods-are-so-popular-right-now/l-intro-1650327494.jpg
    example_title: Fried Food
  - src: >-
      https://www.seriouseats.com/thmb/WzQz05gt5witRGeOYKTcTqfe1gs=/1500x0/filters:no_upscale():max_bytes(150000):strip_icc()/butter-basted-pan-seared-steaks-recipe-hero-06-03b1131c58524be2bd6c9851a2fbdbc3.jpg
    example_title: Meat
  - src: https://assets3.thrillist.com/v1/image/3097381/1200x600/scale;
    example_title: Seafood
  - src: >-
      https://i0.wp.com/post.healthline.com/wp-content/uploads/2020/03/romaine-lettuce-1296x728-body.jpg?w=1155&h=1528
    example_title: Vegetable-Fruit
co2_eq_emissions:
  emissions: 0.02022758375917015

Food Category Classification

This is a Food Category Image Classifier model that has been trained by Kaludi to recognize 11 different categories of foods, including Bread, Dairy Product, Dessert, Egg, Fried Food, Meat, Noodles-Pasta, Rice, Seafood, Soup, and Vegetable-Fruit. It can accurately classify an image of food into one of these categories by analyzing its visual features. This model can be used by food bloggers, restaurants, and recipe websites to quickly categorize and sort their food images, making it easier to manage their content and provide a better user experience.

Gradio

Tis model supports a Gradio Web UI to run the data-food-classification model: Open In HF Spaces

Validation Metrics

  • Loss: 0.079
  • Accuracy: 0.978
  • Macro F1: 0.978
  • Micro F1: 0.978
  • Weighted F1: 0.978
  • Macro Precision: 0.979
  • Micro Precision: 0.978
  • Weighted Precision: 0.979
  • Macro Recall: 0.978
  • Micro Recall: 0.978
  • Weighted Recall: 0.978