Sync object-detection from metro-analytics-catalog
Browse files- config.json +22 -0
- export_and_quantize.sh +12 -0
config.json
ADDED
|
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_type": "object-detection",
|
| 3 |
+
"library_name": "openvino",
|
| 4 |
+
"pipeline_tag": "object-detection",
|
| 5 |
+
"category": "General Object Detection (80-class COCO)",
|
| 6 |
+
"base_model": "YOLO26 (Ultralytics)",
|
| 7 |
+
"source_framework": "PyTorch (Ultralytics)",
|
| 8 |
+
"supported_precisions": [
|
| 9 |
+
"FP32",
|
| 10 |
+
"FP16",
|
| 11 |
+
"INT8"
|
| 12 |
+
],
|
| 13 |
+
"inference_engine": "OpenVINO",
|
| 14 |
+
"hardware": [
|
| 15 |
+
"CPU",
|
| 16 |
+
"GPU",
|
| 17 |
+
"NPU"
|
| 18 |
+
],
|
| 19 |
+
"detected_classes": "All 80 COCO classes",
|
| 20 |
+
"hf_repo_id": "Intel/object-detection",
|
| 21 |
+
"description": "General-purpose object detection use case using YOLO26 INT8 with all 80 COCO classes"
|
| 22 |
+
}
|
export_and_quantize.sh
CHANGED
|
@@ -46,6 +46,18 @@ if [[ "${APPROVAL}" != "yes" ]]; then
|
|
| 46 |
echo "Download cancelled by user."
|
| 47 |
exit 0
|
| 48 |
fi
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 49 |
echo ""
|
| 50 |
echo "--- Downloading sample test image ---"
|
| 51 |
if [[ ! -f test.jpg ]]; then
|
|
|
|
| 46 |
echo "Download cancelled by user."
|
| 47 |
exit 0
|
| 48 |
fi
|
| 49 |
+
|
| 50 |
+
# Ping the HuggingFace repo to register a tracked download of config.json.
|
| 51 |
+
# This is best-effort: a failure (offline, or before the repo is published)
|
| 52 |
+
# must not stop the export.
|
| 53 |
+
echo "--- Registering HuggingFace download (tracking ping) ---"
|
| 54 |
+
HF_REPO_ID="Intel/object-detection"
|
| 55 |
+
HF_CONFIG_URL="https://huggingface.co/${HF_REPO_ID}/resolve/main/config.json"
|
| 56 |
+
if curl -fsSL -o /dev/null "${HF_CONFIG_URL}"; then
|
| 57 |
+
echo "Registered HuggingFace download for ${HF_REPO_ID}"
|
| 58 |
+
else
|
| 59 |
+
echo "WARNING: HuggingFace tracking ping failed (offline?); continuing." >&2
|
| 60 |
+
fi
|
| 61 |
echo ""
|
| 62 |
echo "--- Downloading sample test image ---"
|
| 63 |
if [[ ! -f test.jpg ]]; then
|