nfsw-classifier-v1 / config.json
filipsedivy's picture
Update config
6a36e01
raw
history blame
217 Bytes
{
"labels": ["Nature", "Other", "Person", "Person half naked", "Person naked"],
"id2label": {"0": "Nature", "1": "Other", "2": "Person", "3": "Person half naked", "4": "Person naked"},
"input_size": [3, 224, 224]
}