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.

Try this model (freshvision)

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