resnet18-food-classifier

hilmansw/resnet18-food-classifier

resnet18-food-classifier is a image classification model from hilmansw. Use it when you need a label for an image. It is set up for transformers. The card lists the license as apache-2.0.

should probably proofread and complete it, then remove this comment. --

Downloads · 30 days

2% of all-time downloads

All-time downloads

780

Public

Repo size

89.6 MB

Likes

Public

Hugging Face

Repo makeup

Click a slice to open those files.

.bin44.8 MB · 100%

At a glance

Task: Image Classification
Library: transformers
License: apache-2.0
Model type: resnet
Access: Public
Created: Oct 14, 2023
Updated: Oct 14, 2023
SHA: df14da9f

Base models

microsoft/resnet-18

Model card

From the Hugging Face model README

Model description

This model is a fine-tuned version of microsoft/resnet-18 on an custom dataset. This model was built using the "Padang Cuisine (Indonesian Food Image Classification)" dataset obtained from Kaggle. During the model building process, this was done using the Pytorch framework with pre-trained Resnet-18. The method used during the process of building this classification model is fine-tuning with the dataset.

Training results

Epoch	Accuracy
1.0	0.6030
2.0	0.8342
3.0	0.8442
4.0	0.8191
5.0	0.8693
6.0	0.8643
7.0	0.8744
8.0	0.8643
9.0	0.8744
10.0	0.8744
11.0	0.8794
12.0	0.8744
13.0	0.8894
14.0	0.8794
15.0	0.8945

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

loss_function = CrossEntropyLoss
optimizer = AdamW
learning_rate: 0.00001
batch_size: 16
num_epochs: 15

Framework versions

Transformers 4.34.0
Pytorch 2.0.1+cu118
Datasets 2.14.5
Tokenizers 0.14.1

Task: Image Classification
Library: transformers
Type: resnet
License: apache-2.0
Created: Oct 14, 2023
Updated: Oct 14, 2023