From e4dc68bc69c9a1e3e56d11e23de9bb432e7bd1d5 Mon Sep 17 00:00:00 2001 From: YYJ-aaaa <1058387004@qq.com> Date: Wed, 6 Nov 2024 10:52:32 +0800 Subject: [PATCH] first commit --- config.json | 42 ++++++++++++++++++++++++++++++++++++++++ model.safetensors | 3 +++ preprocessor_config.json | 15 ++++++++++++++ pytorch_model.bin | 3 +++ 4 files changed, 63 insertions(+) create mode 100644 config.json create mode 100644 model.safetensors create mode 100644 preprocessor_config.json create mode 100644 pytorch_model.bin diff --git a/config.json b/config.json new file mode 100644 index 0000000..438845e --- /dev/null +++ b/config.json @@ -0,0 +1,42 @@ +{ + "_name_or_path": "google/vit-base-patch16-224-in21k", + "architectures": [ + "ViTForImageClassification" + ], + "attention_probs_dropout_prob": 0.0, + "hidden_act": "gelu", + "hidden_dropout_prob": 0.0, + "hidden_size": 768, + "id2label": { + "0": "0-2", + "1": "3-9", + "2": "10-19", + "3": "20-29", + "4": "30-39", + "5": "40-49", + "6": "50-59", + "7": "60-69", + "8": "more than 70" + }, + "image_size": 224, + "initializer_range": 0.02, + "intermediate_size": 3072, + "label2id": { + "0-2": 0, + "3-9": 1, + "10-19": 2, + "20-29": 3, + "30-39": 4, + "40-49": 5, + "50-59": 6, + "60-69": 7, + "more than 70": 8 + }, + "layer_norm_eps": 1e-12, + "model_type": "vit", + "num_attention_heads": 12, + "num_channels": 3, + "num_hidden_layers": 12, + "patch_size": 16, + "transformers_version": "4.5.0.dev0" +} diff --git a/model.safetensors b/model.safetensors new file mode 100644 index 0000000..31269ec --- /dev/null +++ b/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:44813db5205ae64124c5fe19244453d866e9516aa18afca454cc285cfec67262 +size 343245508 diff --git a/preprocessor_config.json b/preprocessor_config.json new file mode 100644 index 0000000..a3a75db --- /dev/null +++ b/preprocessor_config.json @@ -0,0 +1,15 @@ +{ + "do_normalize": true, + "do_resize": true, + "image_mean": [ + 0.5, + 0.5, + 0.5 + ], + "image_std": [ + 0.5, + 0.5, + 0.5 + ], + "size": 224 +} \ No newline at end of file diff --git a/pytorch_model.bin b/pytorch_model.bin new file mode 100644 index 0000000..4a11fe3 --- /dev/null +++ b/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4dc2e2d6b7c0e131f581c46658bafc5e773abe584477a7e2679787f51dbdefa6 +size 343303671