| { | |
| "model_name": "convnext_base.fb_in22k_ft_in1k_384", | |
| "num_classes": 6, | |
| "class_names": [ | |
| "HDPE", | |
| "LDPE", | |
| "Other", | |
| "PET", | |
| "PP", | |
| "PS" | |
| ], | |
| "class_to_idx": { | |
| "HDPE": 0, | |
| "LDPE": 1, | |
| "Other": 2, | |
| "PET": 3, | |
| "PP": 4, | |
| "PS": 5 | |
| }, | |
| "idx_to_class": { | |
| "0": "HDPE", | |
| "1": "LDPE", | |
| "2": "Other", | |
| "3": "PET", | |
| "4": "PP", | |
| "5": "PS" | |
| }, | |
| "img_size": 384, | |
| "imagenet_mean": [ | |
| 0.485, | |
| 0.456, | |
| 0.406 | |
| ], | |
| "imagenet_std": [ | |
| 0.229, | |
| 0.224, | |
| 0.225 | |
| ], | |
| "hierarchy": { | |
| "PET": { | |
| "category": "Rigid Thermoplastic", | |
| "recyclable": true, | |
| "code": 1 | |
| }, | |
| "HDPE": { | |
| "category": "Rigid Thermoplastic", | |
| "recyclable": true, | |
| "code": 2 | |
| }, | |
| "PP": { | |
| "category": "Rigid Thermoplastic", | |
| "recyclable": true, | |
| "code": 5 | |
| }, | |
| "LDPE": { | |
| "category": "Flexible Thermoplastic", | |
| "recyclable": true, | |
| "code": 4 | |
| }, | |
| "PS": { | |
| "category": "Brittle Thermoplastic", | |
| "recyclable": false, | |
| "code": 6 | |
| }, | |
| "Other": { | |
| "category": "Unknown", | |
| "recyclable": null, | |
| "code": 7 | |
| } | |
| }, | |
| "test_accuracy": 0.7801980198019802, | |
| "test_macro_f1": 0.8077352266209736, | |
| "best_val_f1": 0.7676573659411366, | |
| "strategy": "Direct classification (no YOLO), full LLRD, EMA, 5-view TTA" | |
| } |