README.md

metadata

title: Freshvision
emoji: 🐨
colorFrom: yellow
colorTo: yellow
sdk: gradio
sdk_version: 4.42.0
app_file: app.py
pinned: false
license: mit

FreshVision with EfficientNet

FreshVision is a machine learning model to classify freshness for fruits and vegetables. This model is built using EfficientNet. EfficientNet is an image classification model family thas has been trained on more than a million images from the ImageNet database.

Models library

• EfficientNetB0 docs
• EfficientNetB2 docs

Prerequisites

• Python 3
  • torch 2.4.0
  • torchvision 0.19.0
  • tqdm
  • tensorboard

References

• https://pytorch.org/hub/nvidia_deeplearningexamples_efficientnet/
• https://data.mendeley.com/datasets/6ps7gtp2wg/1