|
|
--- |
|
|
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](https://pytorch.org/hub/nvidia_deeplearningexamples_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](https://pytorch.org/vision/main/models/generated/torchvision.models.efficientnet_b0.html) |
|
|
* EfficientNetB2 [docs](https://pytorch.org/vision/main/models/generated/torchvision.models.efficientnet_b2.html#torchvision.models.efficientnet_b2) |
|
|
|
|
|
## 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 |
