Skip to content

Commit 600e379

Browse files
No public description
PiperOrigin-RevId: 765373741
1 parent 3e83611 commit 600e379

File tree

1 file changed

+12
-35
lines changed
  • official/projects/waste_identification_ml

1 file changed

+12
-35
lines changed

official/projects/waste_identification_ml/README.md

Lines changed: 12 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -3,59 +3,36 @@
33
Instance segmentation models for identification of recyclables on conveyor
44
belts.
55

6-
Note: These are demo models built on limited datasets. If youre interested in
7-
updated versions of the models, or in using models trained on specific
8-
materials, reach out to [email protected]
6+
We provide retraining and fine-tuning utilities, but if you're interested in
7+
partnering more closely with us reach out to
8+
99

1010
## Overview
1111

1212
CircularNet is built using Mask RCNN, which is a deep learning model for
1313
instance image segmentation, where the goal is to assign instance level labels
1414
(e.g. person1, person2, cat) to every pixel in an input image.
1515

16-
Mask RCNN algorithm is available in the TensorFlow Model Garden which is a
16+
Mask RCNN algorithm is available in the TensorFlow Model Garden, which is a
1717
repository with a number of different implementations of state-of-the-art models
1818
and modeling solutions for TensorFlow users.
1919

2020
## Model Categories
2121

22-
- Material Type - Identifies the high level material type (e.g. plastic, paper
23-
etc) of an object
24-
- Material Form - Categorizes objects based on the form factor (e.g. cup,
22+
- **Material Type:** Identifies the material type (metal, paper etc) of an
23+
object. For plastic, resin types are also identified (HDPE, PET, LDPE, etc).
24+
- **Material Form:** Categorizes objects based on the form factor (cup,
2525
bottle, bag etc)
26-
- Plastic Type - Identifies the plastic resin type of the object (e.g. PET,
27-
HDPE, LDPE, etc)
26+
- **Example inference label:** Plastics-PET_Bottle
2827

29-
## Model paths in GCP buckets
30-
31-
### 1 Model Strategy (latest model)
32-
### Single unified model that performs material type and material form detections
28+
### Latest model
29+
### Single unified model that performs material type and form detections
3330

3431
Model categories | Model backbone | Model type | GCP bucket path |
3532
| ------ | ------ | ----- | ------ |
3633
Material Type & Form | Resnet | saved model | [click here](https://storage.googleapis.com/tf_model_garden/vision/waste_identification_ml/Jan2025_ver2_merged_1024_1024.zip)
3734

38-
### 2 Model Strategy
39-
### Combines plastic type and material type identifications into a unified model
40-
### v2 version is trained on larger datasets than v1
41-
42-
Model categories | Model backbone | Model type | GCP bucket path |
43-
| ------ | ------ | ----- | ------ |
44-
Material Type Model | Resnet | saved model | [click here](https://storage.googleapis.com/tf_model_garden/vision/waste_identification_ml/two_model_strategy/material/material_version_2.zip)
45-
Material Form Model | Resnet | saved model | [click here](https://storage.googleapis.com/tf_model_garden/vision/waste_identification_ml/two_model_strategy/material_form/material_form_version_2.zip)
46-
Material Type Model V2 | Resnet | saved model | [click here](https://storage.googleapis.com/tf_model_garden/vision/waste_identification_ml/resnet_material_v2.zip)
47-
Material Form Model V2 | Resnet | saved model | [click here](https://storage.googleapis.com/tf_model_garden/vision/waste_identification_ml/resnet_material_form_v2.zip)
48-
Material Type Model V2| MobileNet | saved model | [click here](https://storage.googleapis.com/tf_model_garden/vision/waste_identification_ml/mobilenet_material.zip)
49-
Material Form Model V2| MobileNet | saved model | [click here](https://storage.googleapis.com/tf_model_garden/vision/waste_identification_ml/mobilenet_material_form.zip)
50-
51-
### 3 Model Strategy
52-
53-
| Model categories | Model backbone | Model type | GCP bucket path |
54-
| ------ | ------ | ----- | ------ |
55-
| Material Model | Resnet | saved model & TFLite | [click here](https://storage.googleapis.com/tf_model_garden/vision/waste_identification_ml/material_model.zip) |
56-
| Material Form Model | Resnet | saved model & TFLite | [click here](https://storage.googleapis.com/tf_model_garden/vision/waste_identification_ml/material_form_model.zip) |
57-
|Plastic Model | Resnet | saved model & TFLite | [click here](https://storage.googleapis.com/tf_model_garden/vision/waste_identification_ml/plastic_types_model.zip) |
58-
5935
## Authors and Maintainers
6036
Umair Sabir
61-
Sujit Sanjeev
37+
Sujit Sanjeev
38+
Ethan Steele

0 commit comments

Comments
 (0)