vagheshpatel commited on
Commit
4a46c88
·
verified ·
1 Parent(s): f687e5f

Sync motion-tracking from metro-analytics-catalog

Browse files
Files changed (1) hide show
  1. export_and_quantize.sh +24 -2
export_and_quantize.sh CHANGED
@@ -33,11 +33,22 @@ fi
33
 
34
  echo "--- Installing dependencies ---"
35
  if [[ "${PRECISION}" == "INT8" ]]; then
36
- pip install -qU "openvino>=2026.0.0" "nncf>=3.0.0" ultralytics
37
  else
38
- pip install -qU "openvino>=2026.0.0" ultralytics
39
  fi
40
 
 
 
 
 
 
 
 
 
 
 
 
41
  echo "--- Downloading sample test video ---"
42
  if [[ ! -f test_video.mp4 ]]; then
43
  wget -q -O test_video.mp4 \
@@ -47,6 +58,17 @@ else
47
  echo "Already present: test_video.mp4"
48
  fi
49
 
 
 
 
 
 
 
 
 
 
 
 
50
  echo "--- Downloading sample test image ---"
51
  if [[ ! -f test.jpg ]]; then
52
  wget -q -O test.jpg https://ultralytics.com/images/bus.jpg
 
33
 
34
  echo "--- Installing dependencies ---"
35
  if [[ "${PRECISION}" == "INT8" ]]; then
36
+ pip install -qU openvino nncf ultralytics
37
  else
38
+ pip install -qU openvino ultralytics
39
  fi
40
 
41
+ # Ask for approval before downloading models and sample files
42
+ echo ""
43
+ echo "This script will download:"
44
+ echo " - Model weights and/or sample files"
45
+ echo ""
46
+ read -p "Continue with downloads? (yes/no): " APPROVAL
47
+ if [[ "${APPROVAL}" != "yes" ]]; then
48
+ echo "Download cancelled by user."
49
+ exit 0
50
+ fi
51
+ echo ""
52
  echo "--- Downloading sample test video ---"
53
  if [[ ! -f test_video.mp4 ]]; then
54
  wget -q -O test_video.mp4 \
 
58
  echo "Already present: test_video.mp4"
59
  fi
60
 
61
+ # Ask for approval before downloading models and sample files
62
+ echo ""
63
+ echo "This script will download:"
64
+ echo " - Model weights and/or sample files"
65
+ echo ""
66
+ read -p "Continue with downloads? (yes/no): " APPROVAL
67
+ if [[ "${APPROVAL}" != "yes" ]]; then
68
+ echo "Download cancelled by user."
69
+ exit 0
70
+ fi
71
+ echo ""
72
  echo "--- Downloading sample test image ---"
73
  if [[ ! -f test.jpg ]]; then
74
  wget -q -O test.jpg https://ultralytics.com/images/bus.jpg