chore: use new nsfw classifier model
This commit is contained in:
@@ -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.
@@ -1,3 +0,0 @@
|
|||||||
version https://git-lfs.github.com/spec/v1
|
|
||||||
oid sha256:7b81155313d894bba3a3b9bace059a6da2a0c509d10cea43fcb0b3c5a4edd26d
|
|
||||||
size 22527617
|
|
||||||
Reference in New Issue
Block a user