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",
|
||||
"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.
@@ -1,3 +0,0 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:7b81155313d894bba3a3b9bace059a6da2a0c509d10cea43fcb0b3c5a4edd26d
|
||||
size 22527617
|
||||
Reference in New Issue
Block a user