diff --git a/.gitattributes b/.gitattributes index a66ec96..22e7699 100644 --- a/.gitattributes +++ b/.gitattributes @@ -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 diff --git a/Models/nsfw-classifier/config.json b/Models/nsfw-classifier/config.json new file mode 100644 index 0000000..9e7f94a --- /dev/null +++ b/Models/nsfw-classifier/config.json @@ -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" +} diff --git a/Models/nsfw-classifier/model.onnx b/Models/nsfw-classifier/model.onnx new file mode 100644 index 0000000..e59a30e --- /dev/null +++ b/Models/nsfw-classifier/model.onnx @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7b81155313d894bba3a3b9bace059a6da2a0c509d10cea43fcb0b3c5a4edd26d +size 22527617