filipsedivy
/

nfsw-classifier-v1

Image Classification

Model card Files Files and versions Metrics Training metrics Community

filipsedivy commited on Mar 17, 2022

Commit

6a36e01

•

1 Parent(s): 33715d5

Update config

Files changed (1) hide show

config.json +2 -1

config.json CHANGED Viewed

@@ -1,4 +1,5 @@
 {
 	"labels": ["Nature", "Other", "Person", "Person half naked", "Person naked"],
-	"id2label": {"0": "Nature", "1": "Other", "2": "Person", "3": "Person half naked", "4": "Person naked"}
 }

 {
 	"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]
 }