Update README.md
Browse files
README.md
CHANGED
|
@@ -15,7 +15,7 @@ license: mit
|
|
| 15 |
This model is a garbage classification system built on the ResNet50 architecture, fine-tuned for classifying household garbage into 12 distinct categories: paper, cardboard, biological, metal, plastic, green-glass, brown-glass, white-glass, clothes, shoes, batteries, and trash. The model was trained on a dataset sourced from the Kaggle Garbage Classification Dataset by Mostafa Abla. The purpose of this model is to assist in sorting household waste for better recycling efficiency.
|
| 16 |
|
| 17 |
## Model Architecture
|
| 18 |
-
Base Model: ResNet50 (pre-trained on ImageNet)
|
| 19 |
Modifications: The final fully connected layer was modified to output 12 classes instead of the default 1000 classes used in ImageNet.
|
| 20 |
|
| 21 |
## Intended Use
|
|
|
|
| 15 |
This model is a garbage classification system built on the ResNet50 architecture, fine-tuned for classifying household garbage into 12 distinct categories: paper, cardboard, biological, metal, plastic, green-glass, brown-glass, white-glass, clothes, shoes, batteries, and trash. The model was trained on a dataset sourced from the Kaggle Garbage Classification Dataset by Mostafa Abla. The purpose of this model is to assist in sorting household waste for better recycling efficiency.
|
| 16 |
|
| 17 |
## Model Architecture
|
| 18 |
+
Base Model: ResNet50 (pre-trained on ImageNet).
|
| 19 |
Modifications: The final fully connected layer was modified to output 12 classes instead of the default 1000 classes used in ImageNet.
|
| 20 |
|
| 21 |
## Intended Use
|