chore: add nsfw classifier model

2026-06-13 11:27:54 +08:00
parent 6d2c05ec40
commit 5a27317dd9
3 changed files with 40 additions and 0 deletions
@@ -1,3 +1,4 @@
 *.png filter=lfs diff=lfs merge=lfs -text
 *.jpg filter=lfs diff=lfs merge=lfs -text
 *.ttf filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
@@ -0,0 +1,36 @@
+{
+  "architecture": "vit_tiny_patch16_384",
+  "classifier": "head",
+  "crop_mode": "center",
+  "crop_pct": 1.0,
+  "custom_load": true,
+  "export_model_type": "vit_tiny_patch16_384",
+  "first_conv": "patch_embed.proj",
+  "fixed_input_size": true,
+  "hf_hub_id": "Marqo/nsfw-image-detection-384",
+  "input_size": [
+    3,
+    384,
+    384
+  ],
+  "interpolation": "bicubic",
+  "label_names": [
+    "NSFW",
+    "SFW"
+  ],
+  "mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "num_classes": 2,
+  "pool_size": null,
+  "source": "hf-hub",
+  "std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "tag": "augreg_in21k_ft_in1k",
+  "transformers_version": "4.51.3"
+}
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7b81155313d894bba3a3b9bace059a6da2a0c509d10cea43fcb0b3c5a4edd26d
+size 22527617