Update preprocessor_config.json
Browse files- preprocessor_config.json +16 -1
preprocessor_config.json
CHANGED
|
@@ -4,5 +4,20 @@
|
|
| 4 |
"size": 224,
|
| 5 |
"do_normalize": true,
|
| 6 |
"image_mean": [0.5, 0.5, 0.5],
|
| 7 |
-
"image_std": [0.5, 0.5, 0.5]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
}
|
|
|
|
| 4 |
"size": 224,
|
| 5 |
"do_normalize": true,
|
| 6 |
"image_mean": [0.5, 0.5, 0.5],
|
| 7 |
+
"image_std": [0.5, 0.5, 0.5],
|
| 8 |
+
"num_labels": 5,
|
| 9 |
+
"id2label": {
|
| 10 |
+
"0": "stable_diffusion",
|
| 11 |
+
"1": "midjourney",
|
| 12 |
+
"2": "dalle",
|
| 13 |
+
"3": "real",
|
| 14 |
+
"4": "other_ai"
|
| 15 |
+
},
|
| 16 |
+
"label2id": {
|
| 17 |
+
"stable_diffusion": 0,
|
| 18 |
+
"midjourney": 1,
|
| 19 |
+
"dalle": 2,
|
| 20 |
+
"real": 3,
|
| 21 |
+
"other_ai": 4
|
| 22 |
+
}
|
| 23 |
}
|