| { | |
| "model_type": "resnet18", | |
| "num_classes": 10, | |
| "class_names": [ | |
| "battery", | |
| "biological", | |
| "cardboard", | |
| "clothes", | |
| "glass", | |
| "metal", | |
| "paper", | |
| "plastic", | |
| "shoes", | |
| "trash" | |
| ], | |
| "task": "image-classification", | |
| "dataset": "Garbage Classification v2 (Kaggle)" | |
| } |