Update README.md
Browse files
README.md
CHANGED
@@ -51,14 +51,16 @@ model-index:
|
|
51 |
|
52 |
## Overview
|
53 |
This model takes images or video frames as input, and identifies the most likely types of trash present in the scene.
|
54 |
-
The model has been specifically built for aquatic trash, but performs almost equally well on terrestrial trash.
|
|
|
55 |
|
56 |
# Usage
|
57 |
The model has been trained on 120 x 120 RGB images. To evaluate the contents of an image, you will need to pass in a tensor of shape (120,120,3). <br>
|
58 |
Output consists of a 10-d tensor of class probabilities.
|
59 |
|
60 |
## Training and Classes
|
61 |
-
Trained for 22 epochs on
|
|
|
62 |
|
63 |
#### Class labels
|
64 |
trash_classes = ['battery','biological','glass','cardboard','clothes','metal','paper','plastic','shoes','trash']
|
@@ -83,7 +85,8 @@ class_to_idx = {<br>
|
|
83 |
|
84 |
## Limitations
|
85 |
The model has limited training data of trash in the environment. Additionally, the model overrepresents plastic and glass
|
86 |
-
in its predictions due to sampling bias and visual similarities between plastic, glass, and other common types of trash.
|
87 |
-
|
|
|
88 |
|
89 |
-
|
|
|
51 |
|
52 |
## Overview
|
53 |
This model takes images or video frames as input, and identifies the most likely types of trash present in the scene.
|
54 |
+
The model has been specifically built for aquatic trash, but performs almost equally well on terrestrial trash.
|
55 |
+
Applications include automatic trash classification, ecological monitoring, and sorting at recycling plants.
|
56 |
|
57 |
# Usage
|
58 |
The model has been trained on 120 x 120 RGB images. To evaluate the contents of an image, you will need to pass in a tensor of shape (120,120,3). <br>
|
59 |
Output consists of a 10-d tensor of class probabilities.
|
60 |
|
61 |
## Training and Classes
|
62 |
+
Trained for 22 epochs on 3000 data points. Model accuracies are in the sidebar.<br>
|
63 |
+
Please read the 'Limitations' section for information on how the model was evaluated for accuracy.
|
64 |
|
65 |
#### Class labels
|
66 |
trash_classes = ['battery','biological','glass','cardboard','clothes','metal','paper','plastic','shoes','trash']
|
|
|
85 |
|
86 |
## Limitations
|
87 |
The model has limited training data of trash in the environment. Additionally, the model overrepresents plastic and glass
|
88 |
+
in its predictions due to sampling bias and visual similarities between plastic, glass, and other common types of trash.
|
89 |
+
Additionally, may types of trash look visually similar or identical, even to humans.
|
90 |
+
As a solution, the model is marked as 'correct' when the correct label is within the model's top r most predicted trash types.
|
91 |
|
92 |
+
r = 3 gives the most appropriate results.
|