etemkocaaslan commited on
Commit
cafce36
·
verified ·
1 Parent(s): 2f63147

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +110 -0
README.md ADDED
@@ -0,0 +1,110 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: agpl-3.0
3
+ library_name: ultralytics
4
+ pipeline_tag: object-detection
5
+ tags:
6
+ - yolo
7
+ - yolo11
8
+ - yolo26
9
+ - object-detection
10
+ - worker-detection
11
+ - person-detection
12
+ - industrial
13
+ - safety
14
+ - computer-vision
15
+ base_model:
16
+ - Ultralytics/YOLO11
17
+ ---
18
+
19
+ # BalkonTech Models — Factory Worker Detection
20
+
21
+ Fine-tuned YOLO models for detecting **factory workers** in industrial environments. These models were trained on real-world factory footage to reliably localize workers under challenging conditions such as occlusion, machinery clutter, and varied lighting.
22
+
23
+ ## Models
24
+
25
+ | File | Base model | Size | Task |
26
+ |---|---|---|---|
27
+ | `yolo11x_best.pt` | YOLO11x | 114 MB | Worker detection |
28
+ | `yolo26x_best.pt` | YOLO26x | 118 MB | Worker detection |
29
+
30
+ **Classes:** `worker` (person in a factory/industrial setting)
31
+
32
+ ## Intended Use
33
+
34
+ - Worker presence detection on the factory floor
35
+ - Occupancy and zone-monitoring analytics
36
+ - Input stage for downstream safety systems (e.g., restricted-area alerts)
37
+
38
+ **Out of scope:** These models are not certified safety devices. Do not use them as the sole mechanism for life-critical decisions. Face recognition or identification of individuals is not supported and not intended.
39
+
40
+ ## Usage
41
+
42
+ ```python
43
+ from ultralytics import YOLO
44
+
45
+ # Load either model
46
+ model = YOLO("yolo11x_best.pt") # or "yolo26x_best.pt"
47
+
48
+ # Inference on an image, video, or stream
49
+ results = model.predict("factory_frame.jpg", conf=0.4)
50
+
51
+ for r in results:
52
+ for box in r.boxes:
53
+ print(box.cls, box.conf, box.xyxy)
54
+ ```
55
+
56
+ Download directly from the Hub:
57
+
58
+ ```python
59
+ from huggingface_hub import hf_hub_download
60
+ from ultralytics import YOLO
61
+
62
+ weights = hf_hub_download(
63
+ repo_id="etemkocaaslan/balkontech-models",
64
+ filename="yolo11x_best.pt",
65
+ )
66
+ model = YOLO(weights)
67
+ ```
68
+
69
+ ## Training
70
+
71
+ - **Base models:** Ultralytics YOLO11x and YOLO26x pretrained weights
72
+ - **Data:** Proprietary dataset of factory-floor imagery annotated for workers
73
+ - **Fine-tuning:** Standard Ultralytics training pipeline
74
+
75
+ <!-- TODO: add training details (epochs, image size, augmentations, dataset size) -->
76
+
77
+ ## Evaluation
78
+
79
+ <!-- TODO: fill in validation metrics -->
80
+
81
+ | Model | mAP50 | mAP50-95 | Precision | Recall |
82
+ |---|---|---|---|---|
83
+ | yolo11x_best | 0.9512 | 0.5209 | 0.9973 | 0.9200 |
84
+ | yolo26x_best | 0.9457 | 0.5291 | 0.9683 | 0.9200 |
85
+
86
+ ## Limitations
87
+
88
+ - Trained on factory environments; performance may degrade in outdoor or non-industrial scenes.
89
+ - Heavy occlusion, unusual poses, or extreme camera angles may reduce recall.
90
+ - Not evaluated for fairness across demographics; detections are class-level only (no identity).
91
+
92
+ ## Ethical Considerations
93
+
94
+ These models detect people in workplaces. Deployments should comply with local privacy and labor regulations (e.g., KVKK/GDPR), inform affected workers, and avoid use for individual surveillance or performance tracking.
95
+
96
+ ## License
97
+
98
+ Released under **AGPL-3.0**, consistent with the [Ultralytics license](https://www.ultralytics.com/legal/agpl-3-0-software-license) of the base models. For commercial licensing of Ultralytics-derived models, see [Ultralytics Licensing](https://www.ultralytics.com/license).
99
+
100
+ ## Citation
101
+
102
+ ```bibtex
103
+ @misc{balkontech-worker-detection,
104
+ author = {Kocaaslan, Etem},
105
+ title = {BalkonTech Models: Fine-tuned YOLO for Factory Worker Detection},
106
+ year = {2026},
107
+ publisher = {Hugging Face},
108
+ howpublished = {\url{https://huggingface.co/etemkocaaslan/balkontech-models}}
109
+ }
110
+ ```