aculotta commited on
Commit
8303a94
1 Parent(s): 56e850a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -5
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 3500 data points.
 
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
- As a solution, the model is marked as 'correct' when the correct label is within the model's top N most predicted trash types.
 
88
 
89
- n = 3 gives the most appropriate results.
 
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.