diff --git a/config.json b/config.json new file mode 100644 index 0000000..7ff9b90 --- /dev/null +++ b/config.json @@ -0,0 +1,41 @@ +{ + "_name_or_path": "google/vit-base-patch16-224-in21k", + "architectures": [ + "ViTForImageClassification" + ], + "attention_probs_dropout_prob": 0.0, + "encoder_stride": 16, + "hidden_act": "gelu", + "hidden_dropout_prob": 0.0, + "hidden_size": 768, + "id2label": { + "0": "benign_keratosis-like_lesions", + "1": "basal_cell_carcinoma", + "2": "actinic_keratoses", + "3": "vascular_lesions", + "4": "melanocytic_Nevi", + "5": "melanoma", + "6": "dermatofibroma" + }, + "image_size": 224, + "initializer_range": 0.02, + "intermediate_size": 3072, + "label2id": { + "benign_keratosis-like_lesions": 0, + "basal_cell_carcinoma": 1, + "actinic_keratoses": 2, + "vascular_lesions": 3, + "melanocytic_Nevi": 4, + "melanoma": 5, + "dermatofibroma": 6 + }, + "layer_norm_eps": 1e-12, + "model_type": "vit", + "num_attention_heads": 12, + "num_channels": 3, + "num_hidden_layers": 12, + "patch_size": 16, + "qkv_bias": true, + "torch_dtype": "float32", + "transformers_version": "4.38.2" +} \ No newline at end of file diff --git a/model.safetensors b/model.safetensors new file mode 100644 index 0000000..59b50b7 --- /dev/null +++ b/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b8136bbbdd085cc97f69bdef01bd0c4c54262cf3ba44f5a94ebc9c8c54e85d60 +size 343239356 diff --git a/preprocessor_config.json b/preprocessor_config.json new file mode 100644 index 0000000..e0cbd0e --- /dev/null +++ b/preprocessor_config.json @@ -0,0 +1,41 @@ +{ + "_name_or_path": "google/vit-base-patch16-224-in21k", + "architectures": [ + "ViTForImageClassification" + ], + "attention_probs_dropout_prob": 0.0, + "encoder_stride": 16, + "hidden_act": "gelu", + "hidden_dropout_prob": 0.0, + "hidden_size": 768, + "id2label": { + "0": "benign_keratosis-like_lesions", + "1": "basal_cell_carcinoma", + "2": "actinic_keratoses", + "3": "vascular_lesions", + "4": "melanocytic_Nevi", + "5": "melanoma", + "6": "dermatofibroma" + }, + "image_size": 224, + "initializer_range": 0.02, + "intermediate_size": 3072, + "label2id": { + "benign_keratosis-like_lesions": 0, + "basal_cell_carcinoma": 1, + "actinic_keratoses": 2, + "vascular_lesions": 3, + "melanocytic_Nevi": 4, + "melanoma": 5, + "dermatofibroma": 6 + }, + "layer_norm_eps": 1e-12, + "model_type": "vit", + "num_attention_heads": 12, + "num_channels": 3, + "num_hidden_layers": 12, + "patch_size": 16, + "qkv_bias": true, + "torch_dtype": "float32", + "transformers_version": "4.38.2" +}