|
3 | 3 | Instance segmentation models for identification of recyclables on conveyor
|
4 | 4 | belts.
|
5 | 5 |
|
6 |
| -Note: These are demo models built on limited datasets. If you’re 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 | + |
9 | 9 |
|
10 | 10 | ## Overview
|
11 | 11 |
|
12 | 12 | CircularNet is built using Mask RCNN, which is a deep learning model for
|
13 | 13 | instance image segmentation, where the goal is to assign instance level labels
|
14 | 14 | (e.g. person1, person2, cat) to every pixel in an input image.
|
15 | 15 |
|
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 |
17 | 17 | repository with a number of different implementations of state-of-the-art models
|
18 | 18 | and modeling solutions for TensorFlow users.
|
19 | 19 |
|
20 | 20 | ## Model Categories
|
21 | 21 |
|
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, |
25 | 25 | 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 |
28 | 27 |
|
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 |
33 | 30 |
|
34 | 31 | Model categories | Model backbone | Model type | GCP bucket path |
|
35 | 32 | | ------ | ------ | ----- | ------ |
|
36 | 33 | 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)
|
37 | 34 |
|
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 |
| - |
59 | 35 | ## Authors and Maintainers
|
60 | 36 | Umair Sabir
|
61 |
| -Sujit Sanjeev |
| 37 | +Sujit Sanjeev |
| 38 | +Ethan Steele |
0 commit comments