diff --git a/config.json b/config.json new file mode 100644 index 0000000..cf4a55a --- /dev/null +++ b/config.json @@ -0,0 +1,42 @@ +{ + "_name_or_path": "dima806/facial_emotions_image_detection", + "architectures": [ + "ViTForImageClassification" + ], + "attention_probs_dropout_prob": 0.0, + "encoder_stride": 16, + "hidden_act": "gelu", + "hidden_dropout_prob": 0.0, + "hidden_size": 768, + "id2label": { + "0": "sad", + "1": "disgust", + "2": "angry", + "3": "neutral", + "4": "fear", + "5": "surprise", + "6": "happy" + }, + "image_size": 224, + "initializer_range": 0.02, + "intermediate_size": 3072, + "label2id": { + "angry": 2, + "disgust": 1, + "fear": 4, + "happy": 6, + "neutral": 3, + "sad": 0, + "surprise": 5 + }, + "layer_norm_eps": 1e-12, + "model_type": "vit", + "num_attention_heads": 12, + "num_channels": 3, + "num_hidden_layers": 12, + "patch_size": 16, + "problem_type": "single_label_classification", + "qkv_bias": true, + "torch_dtype": "float32", + "transformers_version": "4.39.3" +} diff --git a/model.safetensors b/model.safetensors new file mode 100644 index 0000000..3542a97 --- /dev/null +++ b/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ffad6b9e5cb20d4857f4b532c2543055d7d0e7e1422ab7469087edaab82f8dd8 +size 343239356 diff --git a/preprocessor_config.json b/preprocessor_config.json new file mode 100644 index 0000000..479edce --- /dev/null +++ b/preprocessor_config.json @@ -0,0 +1,36 @@ +{ + "_valid_processor_keys": [ + "images", + "do_resize", + "size", + "resample", + "do_rescale", + "rescale_factor", + "do_normalize", + "image_mean", + "image_std", + "return_tensors", + "data_format", + "input_data_format" + ], + "do_normalize": true, + "do_rescale": true, + "do_resize": true, + "image_mean": [ + 0.5, + 0.5, + 0.5 + ], + "image_processor_type": "ViTImageProcessor", + "image_std": [ + 0.5, + 0.5, + 0.5 + ], + "resample": 2, + "rescale_factor": 0.00392156862745098, + "size": { + "height": 224, + "width": 224 + } +} diff --git a/pytorch_model.bin b/pytorch_model.bin new file mode 100644 index 0000000..223e03a --- /dev/null +++ b/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fcebce67a9ad54f8778a0274d31f904af73025f3eaaf8301f47920f7db29a567 +size 343284077 diff --git a/training_args.bin b/training_args.bin new file mode 100644 index 0000000..ecc5e79 --- /dev/null +++ b/training_args.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:40b9d6436acc2bf7357bd6544c1b5ac091ba87da7d2f748d85a8e993b83d4108 +size 4411