Duplicate from prithivMLmods/Geometric-Shapes-Classification

Co-authored-by: Prithiv Sakthi <prithivMLmods@users.noreply.huggingface.co>

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,124 @@
+---
+license: apache-2.0
+datasets:
+- prithivMLmods/Math-Shapes
+language:
+- en
+base_model:
+- google/siglip2-base-patch16-224
+pipeline_tag: image-classification
+library_name: transformers
+tags:
+- Shapes
+- Geometric
+- SigLIP2
+- art
+---
+
+![zdfdf.png](https://cdn-uploads.huggingface.co/production/uploads/65bb837dbfb878f46c77de4c/z98n2b4kPFRoxPGKddrno.png)
+  
+# **Geometric-Shapes-Classification**
+
+> **Geometric-Shapes-Classification** is an image classification vision-language encoder model fine-tuned from **google/siglip2-base-patch16-224** for a multi-class shape recognition task. It classifies various geometric shapes using the **SiglipForImageClassification** architecture.
+
+```py
+Classification Report:
+                 precision    recall  f1-score   support
+
+       Circle ◯     0.9921    0.9987    0.9953      1500
+         Kite ⬰     0.9927    0.9927    0.9927      1500
+Parallelogram ▰     0.9926    0.9840    0.9883      1500
+    Rectangle ▭     0.9993    0.9913    0.9953      1500
+      Rhombus ◆     0.9846    0.9820    0.9833      1500
+       Square ◼     0.9914    0.9987    0.9950      1500
+    Trapezoid ⏢     0.9966    0.9793    0.9879      1500
+     Triangle ▲     0.9772    0.9993    0.9881      1500
+
+       accuracy                         0.9908     12000
+      macro avg     0.9908    0.9908    0.9907     12000
+   weighted avg     0.9908    0.9908    0.9907     12000
+```
+
+![download (3).png](https://cdn-uploads.huggingface.co/production/uploads/65bb837dbfb878f46c77de4c/WAdeb9cy5DBb-zZ0TPx0k.png)
+
+The model categorizes images into the following classes:
+
+- **Class 0:** Circle ◯  
+- **Class 1:** Kite ⬰  
+- **Class 2:** Parallelogram ▰  
+- **Class 3:** Rectangle ▭  
+- **Class 4:** Rhombus ◆  
+- **Class 5:** Square ◼  
+- **Class 6:** Trapezoid ⏢  
+- **Class 7:** Triangle ▲  
+
+---
+
+# **Run with Transformers 🤗**
+
+```python
+!pip install -q transformers torch pillow gradio
+```
+
+```python
+import gradio as gr
+from transformers import AutoImageProcessor
+from transformers import SiglipForImageClassification
+from PIL import Image
+import torch
+
+# Load model and processor
+model_name = "prithivMLmods/Geometric-Shapes-Classification"
+model = SiglipForImageClassification.from_pretrained(model_name)
+processor = AutoImageProcessor.from_pretrained(model_name)
+
+# Label mapping with symbols
+labels = {
+    "0": "Circle ◯",
+    "1": "Kite ⬰",
+    "2": "Parallelogram ▰",
+    "3": "Rectangle ▭",
+    "4": "Rhombus ◆",
+    "5": "Square ◼",
+    "6": "Trapezoid ⏢",
+    "7": "Triangle ▲"
+}
+
+def classify_shape(image):
+    """Classifies the geometric shape in the input image."""
+    image = Image.fromarray(image).convert("RGB")
+    inputs = processor(images=image, return_tensors="pt")
+
+    with torch.no_grad():
+        outputs = model(**inputs)
+        logits = outputs.logits
+        probs = torch.nn.functional.softmax(logits, dim=1).squeeze().tolist()
+
+    predictions = {labels[str(i)]: round(probs[i], 3) for i in range(len(probs))}
+    
+    return predictions
+
+# Gradio interface
+iface = gr.Interface(
+    fn=classify_shape,
+    inputs=gr.Image(type="numpy"),
+    outputs=gr.Label(label="Prediction Scores"),
+    title="Geometric Shapes Classification",
+    description="Upload an image to classify geometric shapes such as circle, triangle, square, and more."
+)
+
+# Launch the app
+if __name__ == "__main__":
+    iface.launch()
+```
+
+---
+
+# **Intended Use**
+
+The **Geometric-Shapes-Classification** model is designed to recognize basic geometric shapes in images. Example use cases:
+
+- **Educational Tools:** For learning and teaching geometry visually.  
+- **Computer Vision Projects:** As a shape detector in robotics or automation.  
+- **Image Analysis:** Recognizing symbols in diagrams or engineering drafts.  
+- **Assistive Technology:** Supporting shape identification for visually impaired applications.

checkpoint-375/config.json

@@ -0,0 +1,58 @@
+{
+  "architectures": [
+    "SiglipForImageClassification"
+  ],
+  "id2label": {
+    "0": "Circle \u25ef",
+    "1": "Kite \u2b30",
+    "2": "Parallelogram \u25b0",
+    "3": "Rectangle \u25ad",
+    "4": "Rhombus \u25c6",
+    "5": "Square \u25fc",
+    "6": "Trapezoid \u23e2",
+    "7": "Triangle \u25b2"
+  },
+  "initializer_factor": 1.0,
+  "label2id": {
+    "Circle \u25ef": 0,
+    "Kite \u2b30": 1,
+    "Parallelogram \u25b0": 2,
+    "Rectangle \u25ad": 3,
+    "Rhombus \u25c6": 4,
+    "Square \u25fc": 5,
+    "Trapezoid \u23e2": 6,
+    "Triangle \u25b2": 7
+  },
+  "model_type": "siglip",
+  "problem_type": "single_label_classification",
+  "text_config": {
+    "attention_dropout": 0.0,
+    "hidden_act": "gelu_pytorch_tanh",
+    "hidden_size": 768,
+    "intermediate_size": 3072,
+    "layer_norm_eps": 1e-06,
+    "max_position_embeddings": 64,
+    "model_type": "siglip_text_model",
+    "num_attention_heads": 12,
+    "num_hidden_layers": 12,
+    "projection_size": 768,
+    "torch_dtype": "float32",
+    "vocab_size": 256000
+  },
+  "torch_dtype": "float32",
+  "transformers_version": "4.50.3",
+  "vision_config": {
+    "attention_dropout": 0.0,
+    "hidden_act": "gelu_pytorch_tanh",
+    "hidden_size": 768,
+    "image_size": 224,
+    "intermediate_size": 3072,
+    "layer_norm_eps": 1e-06,
+    "model_type": "siglip_vision_model",
+    "num_attention_heads": 12,
+    "num_channels": 3,
+    "num_hidden_layers": 12,
+    "patch_size": 16,
+    "torch_dtype": "float32"
+  }
+}

checkpoint-375/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:59774861a5253ee8a2685bef62a6aa8e016c2bc57d6ec515be051100f5ae5f43
+size 371586448

checkpoint-375/optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aa188c598b29736f377f9dffe4bd270589b44c7bc79b653779c972bc68163aab
+size 686592634

checkpoint-375/preprocessor_config.json

@@ -0,0 +1,24 @@
+{
+  "do_convert_rgb": null,
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_processor_type": "SiglipImageProcessor",
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "processor_class": "SiglipProcessor",
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "height": 224,
+    "width": 224
+  }
+}

checkpoint-375/rng_state.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a11027277b6c51f59488df520020d383908a9f170baa7ea4bd0c166aa72cced2
+size 14244

checkpoint-375/scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ac4d1140f5a0513989aca0dc1db73e181db90de91e412469cc8be792cca398dc
+size 1064

checkpoint-375/trainer_state.json

@@ -0,0 +1,44 @@
+{
+  "best_global_step": 375,
+  "best_metric": 0.5122714638710022,
+  "best_model_checkpoint": "siglip2-finetune-full/checkpoint-375",
+  "epoch": 1.0,
+  "eval_steps": 500,
+  "global_step": 375,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.85325,
+      "eval_loss": 0.5122714638710022,
+      "eval_model_preparation_time": 0.0023,
+      "eval_runtime": 142.4763,
+      "eval_samples_per_second": 84.225,
+      "eval_steps_per_second": 10.528,
+      "step": 375
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 750,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 2,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 1.005115956461568e+18,
+  "train_batch_size": 32,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-375/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:206d195b28a5a4e1f61804a2eec06b109a4be9fa8d841f4d552fca404cebe15f
+size 5304

checkpoint-750/config.json

@@ -0,0 +1,58 @@
+{
+  "architectures": [
+    "SiglipForImageClassification"
+  ],
+  "id2label": {
+    "0": "Circle \u25ef",
+    "1": "Kite \u2b30",
+    "2": "Parallelogram \u25b0",
+    "3": "Rectangle \u25ad",
+    "4": "Rhombus \u25c6",
+    "5": "Square \u25fc",
+    "6": "Trapezoid \u23e2",
+    "7": "Triangle \u25b2"
+  },
+  "initializer_factor": 1.0,
+  "label2id": {
+    "Circle \u25ef": 0,
+    "Kite \u2b30": 1,
+    "Parallelogram \u25b0": 2,
+    "Rectangle \u25ad": 3,
+    "Rhombus \u25c6": 4,
+    "Square \u25fc": 5,
+    "Trapezoid \u23e2": 6,
+    "Triangle \u25b2": 7
+  },
+  "model_type": "siglip",
+  "problem_type": "single_label_classification",
+  "text_config": {
+    "attention_dropout": 0.0,
+    "hidden_act": "gelu_pytorch_tanh",
+    "hidden_size": 768,
+    "intermediate_size": 3072,
+    "layer_norm_eps": 1e-06,
+    "max_position_embeddings": 64,
+    "model_type": "siglip_text_model",
+    "num_attention_heads": 12,
+    "num_hidden_layers": 12,
+    "projection_size": 768,
+    "torch_dtype": "float32",
+    "vocab_size": 256000
+  },
+  "torch_dtype": "float32",
+  "transformers_version": "4.50.3",
+  "vision_config": {
+    "attention_dropout": 0.0,
+    "hidden_act": "gelu_pytorch_tanh",
+    "hidden_size": 768,
+    "image_size": 224,
+    "intermediate_size": 3072,
+    "layer_norm_eps": 1e-06,
+    "model_type": "siglip_vision_model",
+    "num_attention_heads": 12,
+    "num_channels": 3,
+    "num_hidden_layers": 12,
+    "patch_size": 16,
+    "torch_dtype": "float32"
+  }
+}

checkpoint-750/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a134585cac12e3065d395f47c8f7a50916ed236e6752f94152d5b1466ed6631c
+size 371586448

checkpoint-750/optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4e7b8773667c165c9b7ea1d9ea7b20c1d7ff46c21735b0201c4284ee44a97853
+size 686592634

checkpoint-750/preprocessor_config.json

@@ -0,0 +1,24 @@
+{
+  "do_convert_rgb": null,
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_processor_type": "SiglipImageProcessor",
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "processor_class": "SiglipProcessor",
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "height": 224,
+    "width": 224
+  }
+}

checkpoint-750/rng_state.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8dfd1f9a2a29539f346868eb36a1b264e5fc13fdb515c6a844bf62e6abee8853
+size 14244

checkpoint-750/scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ab03192a60c4fb6ec6096a859424dfb24171c6b3fc801b38b85ab0fd723b1704
+size 1064

checkpoint-750/trainer_state.json

@@ -0,0 +1,61 @@
+{
+  "best_global_step": 750,
+  "best_metric": 0.03016640432178974,
+  "best_model_checkpoint": "siglip2-finetune-full/checkpoint-750",
+  "epoch": 2.0,
+  "eval_steps": 500,
+  "global_step": 750,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.85325,
+      "eval_loss": 0.5122714638710022,
+      "eval_model_preparation_time": 0.0023,
+      "eval_runtime": 142.4763,
+      "eval_samples_per_second": 84.225,
+      "eval_steps_per_second": 10.528,
+      "step": 375
+    },
+    {
+      "epoch": 1.3333333333333333,
+      "grad_norm": 6.622213840484619,
+      "learning_rate": 7.142857142857143e-05,
+      "loss": 0.4962,
+      "step": 500
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.99075,
+      "eval_loss": 0.03016640432178974,
+      "eval_model_preparation_time": 0.0023,
+      "eval_runtime": 141.9262,
+      "eval_samples_per_second": 84.551,
+      "eval_steps_per_second": 10.569,
+      "step": 750
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 750,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 2,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 2.010231912923136e+18,
+  "train_batch_size": 32,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-750/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:206d195b28a5a4e1f61804a2eec06b109a4be9fa8d841f4d552fca404cebe15f
+size 5304

config.json

@@ -0,0 +1,58 @@
+{
+  "architectures": [
+    "SiglipForImageClassification"
+  ],
+  "id2label": {
+    "0": "Circle \u25ef",
+    "1": "Kite \u2b30",
+    "2": "Parallelogram \u25b0",
+    "3": "Rectangle \u25ad",
+    "4": "Rhombus \u25c6",
+    "5": "Square \u25fc",
+    "6": "Trapezoid \u23e2",
+    "7": "Triangle \u25b2"
+  },
+  "initializer_factor": 1.0,
+  "label2id": {
+    "Circle \u25ef": 0,
+    "Kite \u2b30": 1,
+    "Parallelogram \u25b0": 2,
+    "Rectangle \u25ad": 3,
+    "Rhombus \u25c6": 4,
+    "Square \u25fc": 5,
+    "Trapezoid \u23e2": 6,
+    "Triangle \u25b2": 7
+  },
+  "model_type": "siglip",
+  "problem_type": "single_label_classification",
+  "text_config": {
+    "attention_dropout": 0.0,
+    "hidden_act": "gelu_pytorch_tanh",
+    "hidden_size": 768,
+    "intermediate_size": 3072,
+    "layer_norm_eps": 1e-06,
+    "max_position_embeddings": 64,
+    "model_type": "siglip_text_model",
+    "num_attention_heads": 12,
+    "num_hidden_layers": 12,
+    "projection_size": 768,
+    "torch_dtype": "float32",
+    "vocab_size": 256000
+  },
+  "torch_dtype": "float32",
+  "transformers_version": "4.50.3",
+  "vision_config": {
+    "attention_dropout": 0.0,
+    "hidden_act": "gelu_pytorch_tanh",
+    "hidden_size": 768,
+    "image_size": 224,
+    "intermediate_size": 3072,
+    "layer_norm_eps": 1e-06,
+    "model_type": "siglip_vision_model",
+    "num_attention_heads": 12,
+    "num_channels": 3,
+    "num_hidden_layers": 12,
+    "patch_size": 16,
+    "torch_dtype": "float32"
+  }
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a134585cac12e3065d395f47c8f7a50916ed236e6752f94152d5b1466ed6631c
+size 371586448

preprocessor_config.json

@@ -0,0 +1,24 @@
+{
+  "do_convert_rgb": null,
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_processor_type": "SiglipImageProcessor",
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "processor_class": "SiglipProcessor",
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "height": 224,
+    "width": 224
+  }
+}

training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:206d195b28a5a4e1f61804a2eec06b109a4be9fa8d841f4d552fca404cebe15f
+size 5304