ethz/food101
Viewer • Updated • 101k • 33.1k • 149
How to use mhamza-007/Food-Vision-101 with Keras:
# !pip install -U keras tensorflow huggingface_hub
# Keras needs TensorFlow installed to read "hf://" paths, so the tensorflow backend is selected here;
# "jax" and "torch" also work for computation once TensorFlow is installed.
import os
os.environ["KERAS_BACKEND"] = "tensorflow"
import keras
model = keras.saving.load_model("hf://mhamza-007/Food-Vision-101")
This repository contains a fine-tuned EfficientNetB4 model trained on the Food101 dataset. The Food101 dataset comprises 101 different classes of food, making it an excellent benchmark for image classification tasks in the food domain.
sparse_categorical_crossentropy accuracy | Phase | Loss | Accuracy |
|---|---|---|
| Train | 0.4790 | 87.40% |
| Test | 0.6283 | 79.28% |
Base model
google/efficientnet-b4