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",
"crop_mode": "center",
"crop_mode": "squash",
"crop_pct": 1.0,
"custom_load": true,
"export_model_type": "vit_tiny_patch16_384",
"custom_load": false,
"export_model_type": "eva02_base_patch14_448",
"first_conv": "patch_embed.proj",
"fixed_input_size": true,
"hf_hub_id": "Marqo/nsfw-image-detection-384",
"hf_hub_id": "Freepik/nsfw_image_detector",
"input_size": [
3,
384,
384
448,
448
],
"interpolation": "bicubic",
"label_names": [
"NSFW",
"SFW"
"neutral",
"low",
"medium",
"high"
],
"license": "mit",
"mean": [
0.5,
0.5,
0.5
0.48145466,
0.4578275,
0.40821073
],
"num_classes": 2,
"num_classes": 4,
"pool_size": null,
"source": "hf-hub",
"std": [
0.5,
0.5,
0.5
0.26862954,
0.26130258,
0.27577711
],
"tag": "augreg_in21k_ft_in1k",
"tag": "mim_in22k_ft_in22k_in1k",
"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