akul-29 commited on
Commit
c3ad091
·
verified ·
1 Parent(s): 364c4d1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +41 -1
README.md CHANGED
@@ -7,4 +7,44 @@ pipeline_tag: object-detection
7
  tags:
8
  - planogram
9
  - shelf-gaps
10
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7
  tags:
8
  - planogram
9
  - shelf-gaps
10
+ ---
11
+
12
+ # 🛒 Retail Shelf Gap Detection Model
13
+
14
+ This is a YOLOv8-based object detection model fine-tuned on the **Shelf Images for Planograms** dataset to detect **gaps** in retail shelf arrangements. The model aims to assist in **planogram compliance** checking by identifying empty spaces on shelves and computing compliance scores.
15
+
16
+ ## 📌 Use Case
17
+
18
+ This model is intended for retail analytics teams to:
19
+ - Detect shelf gaps in retail store images.
20
+ - Quantify compliance against a predefined planogram.
21
+ - Evaluate image quality and calculate gap-related metrics.
22
+
23
+ ## 🧠 Model Details
24
+
25
+ - **Base Model:** [Ultralytics/YOLOv8](https://github.com/ultralytics/ultralytics)
26
+ - **Task:** Object Detection
27
+ - **Trained On:** Shelf Images for Planograms Dataset (2095 images)
28
+ - **Framework:** PyTorch via Ultralytics
29
+
30
+ ## 🧪 Metrics and Scoring Breakdown
31
+
32
+ The model is used as part of a retail shelf scoring pipeline that computes:
33
+ - **Gap Score** (50%)
34
+ - **Image Quality Score** (30%)
35
+ - **Gap Density Score** (20%)
36
+ - **Final Compliance Score** = Weighted sum of the above
37
+
38
+ ## 🛠️ How to Use
39
+
40
+ ```python
41
+ from ultralytics import YOLO
42
+ from huggingface_hub import hf_hub_download
43
+
44
+ # Download model
45
+ model_path = hf_hub_download(repo_id="akul-29/Retail-Shelf-Gap-Detection_Model", filename="best.pt")
46
+ model = YOLO(model_path)
47
+
48
+ # Run inference
49
+ results = model("path_to_your_image.jpg")
50
+ results.show()