chore: use new nsfw classifier model

This commit is contained in:
GamerYuan
2026-06-15 20:25:34 +08:00
parent 5a27317dd9
commit 05f4697ace
3 changed files with 23 additions and 20 deletions
+20 -17
View File
@@ -1,36 +1,39 @@
{ {
"architecture": "vit_tiny_patch16_384", "architecture": "eva02_base_patch14_448",
"classifier": "head", "classifier": "head",
"crop_mode": "center", "crop_mode": "squash",
"crop_pct": 1.0, "crop_pct": 1.0,
"custom_load": true, "custom_load": false,
"export_model_type": "vit_tiny_patch16_384", "export_model_type": "eva02_base_patch14_448",
"first_conv": "patch_embed.proj", "first_conv": "patch_embed.proj",
"fixed_input_size": true, "fixed_input_size": true,
"hf_hub_id": "Marqo/nsfw-image-detection-384", "hf_hub_id": "Freepik/nsfw_image_detector",
"input_size": [ "input_size": [
3, 3,
384, 448,
384 448
], ],
"interpolation": "bicubic", "interpolation": "bicubic",
"label_names": [ "label_names": [
"NSFW", "neutral",
"SFW" "low",
"medium",
"high"
], ],
"license": "mit",
"mean": [ "mean": [
0.5, 0.48145466,
0.5, 0.4578275,
0.5 0.40821073
], ],
"num_classes": 2, "num_classes": 4,
"pool_size": null, "pool_size": null,
"source": "hf-hub", "source": "hf-hub",
"std": [ "std": [
0.5, 0.26862954,
0.5, 0.26130258,
0.5 0.27577711
], ],
"tag": "augreg_in21k_ft_in1k", "tag": "mim_in22k_ft_in22k_in1k",
"transformers_version": "4.51.3" "transformers_version": "4.51.3"
} }
Binary file not shown.
-3
View File
@@ -1,3 +0,0 @@
version https://git-lfs.github.com/spec/v1
oid sha256:7b81155313d894bba3a3b9bace059a6da2a0c509d10cea43fcb0b3c5a4edd26d
size 22527617