loss-prevention/check_models.sh at main · intel-retail/loss-prevention · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#!/bin/bash
# Copyright © 2025 Intel Corporation. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

# Script to check if all required models are present

MODELS_DIR="${1:-models}"

# Define required models and their paths
declare -a REQUIRED_MODELS=(
    "object_classification/efficientnet-b0/FP16-INT8/efficientnet-b0.bin"
    "object_classification/efficientnet-b0/FP16-INT8/efficientnet-b0.xml"
    "object_classification/efficientnet-b0/FP32-INT8/efficientnet-b0.bin"
    "object_classification/efficientnet-b0/FP32-INT8/efficientnet-b0.xml"
    "object_classification/efficientnet-b0/efficientnet-b0.json"
    "object_classification/efficientnet-b0/efficientnet-b0.txt"
    "object_classification/face-reidentification-retail-0095/FP16/face-reidentification-retail-0095.bin"
    "object_classification/face-reidentification-retail-0095/FP16/face-reidentification-retail-0095.xml"
    "object_classification/age-gender-recognition-retail-0013/FP16/age-gender-recognition-retail-0013.bin"
    "object_classification/age-gender-recognition-retail-0013/FP16/age-gender-recognition-retail-0013.xml"
    "object_detection/face-detection-retail-0004/FP16/face-detection-retail-0004.bin"
    "object_detection/face-detection-retail-0004/FP16/face-detection-retail-0004.xml"
    "object_detection/yolo11n/FP16/yolo11n.bin"
    "object_detection/yolo11n/FP16/yolo11n.xml"
    "object_detection/yolo11n/FP32/yolo11n.bin"
    "object_detection/yolo11n/FP32/yolo11n.xml"
    "object_detection/yolo11n/INT8/yolo11n.bin"
    "object_detection/yolo11n/INT8/yolo11n.xml"
)

# Check if models directory exists
if [ ! -d "$MODELS_DIR" ]; then
    echo "Models directory not found: $MODELS_DIR"
    exit 0
fi

# Check each required model
MISSING_MODELS=0
for model in "${REQUIRED_MODELS[@]}"; do
    if [ ! -f "$MODELS_DIR/$model" ]; then
        echo "Missing model: $model"
        MISSING_MODELS=1
    fi
done

if [ $MISSING_MODELS -eq 0 ]; then
    echo "✓ All required models are present"
    exit 1
else
    echo "✗ Some models are missing"
    exit 0
fi