vagheshpatel commited on
Commit
430b695
·
verified ·
1 Parent(s): 20bbabf

Sync person-detection from metro-analytics-catalog

Browse files
Files changed (1) hide show
  1. export_and_quantize.sh +15 -2
export_and_quantize.sh CHANGED
@@ -31,11 +31,24 @@ fi
31
 
32
  echo "--- Installing dependencies ---"
33
  if [[ "${PRECISION}" == "INT8" ]]; then
34
- pip install -qU "openvino>=2026.0.0" "nncf>=3.0.0" ultralytics
35
  else
36
- pip install -qU "openvino>=2026.0.0" ultralytics
37
  fi
38
 
 
 
 
 
 
 
 
 
 
 
 
 
 
39
  echo "--- Downloading sample test image ---"
40
  if [[ ! -f test.jpg ]]; then
41
  wget -q -O test.jpg https://ultralytics.com/images/zidane.jpg
 
31
 
32
  echo "--- Installing dependencies ---"
33
  if [[ "${PRECISION}" == "INT8" ]]; then
34
+ pip install -qU openvino nncf ultralytics
35
  else
36
+ pip install -qU openvino ultralytics
37
  fi
38
 
39
+ # Ask for approval before downloading models and sample files
40
+ echo ""
41
+ echo "This script will download:"
42
+ echo " - YOLO26 model weights (if not cached locally)"
43
+ echo " - Sample test image and video files"
44
+ echo ""
45
+ read -p "Continue with downloads? (yes/no): " APPROVAL
46
+ if [[ "${APPROVAL}" != "yes" ]]; then
47
+ echo "Download cancelled by user."
48
+ exit 0
49
+ fi
50
+ echo ""
51
+
52
  echo "--- Downloading sample test image ---"
53
  if [[ ! -f test.jpg ]]; then
54
  wget -q -O test.jpg https://ultralytics.com/images/zidane.jpg