Update README.md
Browse filesadd description dataset
README.md
CHANGED
|
@@ -1,11 +1,25 @@
|
|
| 1 |
-
# 🎯
|
|
|
|
|
|
|
| 2 |
|
| 3 |
-
This is an image classification model trained to classify medical waste into 4 categories, namely cytotoxic, infectious, pathological, and pharmaceutical. The model is based on the Inception v3 architecture and has been adapted to a specific dataset for the task of medical waste classification.
|
| 4 |
|
| 5 |
# 🎯 Model Description
|
| 6 |
|
| 7 |
The model is based on the Inception v3 architecture with modifications to the fully connected layers for adapting it to the specific image classification task. The architecture consists of a feature extractor followed by a global average pooling layer and fully connected layers with ReLU activation and dropout.
|
| 8 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
# 🎯 Usage
|
| 10 |
|
| 11 |
You can use the model that I have saved in pt format as follows:
|
|
@@ -68,3 +82,6 @@ plt.title(f'True Class: {true_label} \n Predicted Class: {predicted_label} (Conf
|
|
| 68 |
plt.show()
|
| 69 |
|
| 70 |
```
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# 🎯 Image Classification Model for Medical Waste Classification
|
| 2 |
+
|
| 3 |
+
This is an image classification model trained to classify medical waste into 4 categories, namely cytotoxic, infectious, pathological, and pharmaceutical. The model is based on the Inception v3 architecture and has been adapted to a specific dataset for the task of medical waste classification.
|
| 4 |
|
|
|
|
| 5 |
|
| 6 |
# 🎯 Model Description
|
| 7 |
|
| 8 |
The model is based on the Inception v3 architecture with modifications to the fully connected layers for adapting it to the specific image classification task. The architecture consists of a feature extractor followed by a global average pooling layer and fully connected layers with ReLU activation and dropout.
|
| 9 |
|
| 10 |
+
# 🎯 Dataset Used
|
| 11 |
+
|
| 12 |
+
The train data for each category is 175 images and the validation data is 50 images.
|
| 13 |
+
The dataset used comes from collecting it myself.
|
| 14 |
+
|
| 15 |
+
# 🎯 Final training results
|
| 16 |
+
|
| 17 |
+
The model gets the following results from training
|
| 18 |
+
- Train Loss: 0.243
|
| 19 |
+
- Val. Losses: 0.252
|
| 20 |
+
- Train Acc: 93.73%
|
| 21 |
+
- Val. Acc: 93.92%
|
| 22 |
+
|
| 23 |
# 🎯 Usage
|
| 24 |
|
| 25 |
You can use the model that I have saved in pt format as follows:
|
|
|
|
| 82 |
plt.show()
|
| 83 |
|
| 84 |
```
|
| 85 |
+
The following are the output results from using this coding
|
| 86 |
+
|
| 87 |
+
|