Update README.md
Browse files
README.md
CHANGED
|
@@ -23,7 +23,7 @@ This model is an image classification model trained to identify different types
|
|
| 23 |
This model is a deep learning model for classifying food images into one of 101 categories from the Food101 dataset. It was trained using TensorFlow and likely employs a transfer learning approach, leveraging the features learned by a model pre-trained on a large dataset like ImageNet. The training process included the use of mixed precision for potentially faster training and reduced memory usage.
|
| 24 |
|
| 25 |
* **Developed by:** `Recompense` Me!
|
| 26 |
-
* **Model type:** Image Classification (
|
| 27 |
* **Language(s) (NLP):** N/A (Image Classification)
|
| 28 |
* **License:** MIT
|
| 29 |
* **Finetuned from model:** EfficienntNetB0
|
|
@@ -101,7 +101,7 @@ Carbon emissions can be estimated using the [Machine Learning Impact calculator]
|
|
| 101 |
|
| 102 |
### Model Architecture and Objective
|
| 103 |
|
| 104 |
-
The model is
|
| 105 |
|
| 106 |
### Compute Infrastructure
|
| 107 |
|
|
@@ -114,7 +114,7 @@ The model was trained using a Tesla T4 GPU on Google Cloud in the us-central reg
|
|
| 114 |
* NumPy
|
| 115 |
* Matplotlib
|
| 116 |
* Scikit-learn
|
| 117 |
-
* Helper functions from `helper_functions.py` (
|
| 118 |
|
| 119 |
---
|
| 120 |
|
|
|
|
| 23 |
This model is a deep learning model for classifying food images into one of 101 categories from the Food101 dataset. It was trained using TensorFlow and likely employs a transfer learning approach, leveraging the features learned by a model pre-trained on a large dataset like ImageNet. The training process included the use of mixed precision for potentially faster training and reduced memory usage.
|
| 24 |
|
| 25 |
* **Developed by:** `Recompense` Me!
|
| 26 |
+
* **Model type:** Image Classification (Transfer Learning with a CNN backbone)
|
| 27 |
* **Language(s) (NLP):** N/A (Image Classification)
|
| 28 |
* **License:** MIT
|
| 29 |
* **Finetuned from model:** EfficienntNetB0
|
|
|
|
| 101 |
|
| 102 |
### Model Architecture and Objective
|
| 103 |
|
| 104 |
+
The model is a fine-tuned convolutional neural network (CNN) classifier.Mixed precision training was used for faster training, a modern CNN architecture compatible with `float16` data types. The objective is to minimize the classification loss (e.g., categorical cross-entropy) to accurately predict the food category given an image.
|
| 105 |
|
| 106 |
### Compute Infrastructure
|
| 107 |
|
|
|
|
| 114 |
* NumPy
|
| 115 |
* Matplotlib
|
| 116 |
* Scikit-learn
|
| 117 |
+
* Helper functions from `helper_functions.py` (for plotting, data handling)
|
| 118 |
|
| 119 |
---
|
| 120 |
|