mae-waste-classifier / config.json
ysfad's picture
Upload config.json with huggingface_hub
ad06b8b verified
{
"model_type": "mae_vit",
"architecture": "ViT-Base",
"num_classes": 9,
"image_size": 224,
"patch_size": 16,
"embed_dim": 768,
"num_heads": 12,
"num_layers": 12,
"pretrained": "mae_pretrain_vit_base",
"finetuned_on": "RealWaste",
"validation_accuracy": 0.9327,
"training_accuracy": 0.9989,
"categories": [
"Cardboard",
"Food Organics",
"Glass",
"Metal",
"Miscellaneous Trash",
"Paper",
"Plastic",
"Textile Trash",
"Vegetation"
],
"disposal_instructions": {
"Cardboard": "Flatten and place in recycling bin. Remove any tape or staples.",
"Food Organics": "Compost in organic waste bin or home composter.",
"Glass": "Rinse and place in glass recycling. Remove lids and caps.",
"Metal": "Rinse aluminum/steel cans and place in recycling bin.",
"Miscellaneous Trash": "Dispose in general waste bin. Cannot be recycled.",
"Paper": "Place clean paper in recycling. Remove plastic windows from envelopes.",
"Plastic": "Check recycling number. Rinse containers before recycling.",
"Textile Trash": "Donate if reusable, otherwise dispose in textile recycling.",
"Vegetation": "Compost in organic waste or use for mulch in garden."
}
}