leafora commited on
Commit
0ade1e4
·
verified ·
1 Parent(s): 9576224

update README

Browse files
Files changed (1) hide show
  1. README.md +20 -3
README.md CHANGED
@@ -1,3 +1,20 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # FreshVision with [EfficientNet](https://pytorch.org/hub/nvidia_deeplearningexamples_efficientnet/)
2
+
3
+ ![tomatoo](https://media.tenor.com/N3Glefqal5AAAAAe/tomato-meme.png)
4
+
5
+ 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.
6
+
7
+ ## Models library
8
+ * EfficientNetB0 [docs](https://pytorch.org/vision/main/models/generated/torchvision.models.efficientnet_b0.html)
9
+ * EfficientNetB2 [docs](https://pytorch.org/vision/main/models/generated/torchvision.models.efficientnet_b2.html#torchvision.models.efficientnet_b2)
10
+
11
+ ## Prerequisites
12
+ * Python 3
13
+ * torch 2.4.0
14
+ * torchvision 0.19.0
15
+ * tqdm
16
+ * tensorboard
17
+
18
+ ## References
19
+ * https://pytorch.org/hub/nvidia_deeplearningexamples_efficientnet/
20
+ * https://data.mendeley.com/datasets/6ps7gtp2wg/1