age-group-classifier / config.json
Sharris's picture
Upload config.json with huggingface_hub
0ef3acb verified
{
"architectures": ["ResNet50V2"],
"model_type": "image-classification",
"task": "image-classification",
"id2label": {
"0": "Youth (0-20)",
"1": "Young Adult (21-40)",
"2": "Middle Age (41-60)",
"3": "Senior (61-80)",
"4": "Elderly (81-100)"
},
"label2id": {
"Youth (0-20)": 0,
"Young Adult (21-40)": 1,
"Middle Age (41-60)": 2,
"Senior (61-80)": 3,
"Elderly (81-100)": 4
},
"num_labels": 5,
"image_size": [224, 224],
"num_channels": 3,
"framework": "tf",
"library_name": "tensorflow"
}