Sync worker-safety-detection from metro-analytics-catalog
Browse files- config.json +20 -0
- 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 ---"
|