vagheshpatel commited on
Commit
ac08bdd
·
verified ·
1 Parent(s): d5a5330

Sync worker-safety-detection from metro-analytics-catalog

Browse files
Files changed (2) hide show
  1. config.json +20 -0
  2. export_and_quantize.sh +12 -0
config.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "worker-safety-detection",
3
+ "library_name": "openvino",
4
+ "pipeline_tag": "object-detection",
5
+ "category": "Object Detection (PPE / Safety Compliance)",
6
+ "base_model": "Worker Safety Gear Detection (Intel Edge AI Resources, Geti-trained)",
7
+ "source_framework": "Intel Geti (OpenVINO IR)",
8
+ "supported_precisions": [
9
+ "FP32"
10
+ ],
11
+ "inference_engine": "OpenVINO",
12
+ "hardware": [
13
+ "CPU",
14
+ "GPU",
15
+ "NPU"
16
+ ],
17
+ "detected_classes": "`safety_jacket` (class 0), `safety_helmet` (class 1)",
18
+ "hf_repo_id": "Intel/worker-safety-detection",
19
+ "description": "Worker safety compliance detection using Intel Edge AI Resources Geti-trained INT8 PPE model"
20
+ }
export_and_quantize.sh CHANGED
@@ -29,6 +29,18 @@ if [[ "${APPROVAL}" != "yes" ]]; then
29
  echo "Download cancelled by user."
30
  exit 0
31
  fi
 
 
 
 
 
 
 
 
 
 
 
 
32
  echo ""
33
 
34
  echo "--- Downloading sample test video ---"
 
29
  echo "Download cancelled by user."
30
  exit 0
31
  fi
32
+
33
+ # Ping the HuggingFace repo to register a tracked download of config.json.
34
+ # This is best-effort: a failure (offline, or before the repo is published)
35
+ # must not stop the export.
36
+ echo "--- Registering HuggingFace download (tracking ping) ---"
37
+ HF_REPO_ID="Intel/worker-safety-detection"
38
+ HF_CONFIG_URL="https://huggingface.co/${HF_REPO_ID}/resolve/main/config.json"
39
+ if curl -fsSL -o /dev/null "${HF_CONFIG_URL}"; then
40
+ echo "Registered HuggingFace download for ${HF_REPO_ID}"
41
+ else
42
+ echo "WARNING: HuggingFace tracking ping failed (offline?); continuing." >&2
43
+ fi
44
  echo ""
45
 
46
  echo "--- Downloading sample test video ---"