vagheshpatel commited on
Commit
9da8ed7
·
verified ·
1 Parent(s): 97da892

Sync object-detection from metro-analytics-catalog

Browse files
Files changed (1) hide show
  1. export_and_quantize.sh +24 -2
export_and_quantize.sh CHANGED
@@ -31,11 +31,22 @@ 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/bus.jpg
@@ -44,6 +55,17 @@ else
44
  echo "Already present: test.jpg"
45
  fi
46
 
 
 
 
 
 
 
 
 
 
 
 
47
  echo "--- Downloading sample test video ---"
48
  if [[ ! -f test_video.mp4 ]]; then
49
  wget -q -O test_video.mp4 \
 
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 " - Model weights and/or sample files"
43
+ echo ""
44
+ read -p "Continue with downloads? (yes/no): " APPROVAL
45
+ 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
52
  wget -q -O test.jpg https://ultralytics.com/images/bus.jpg
 
55
  echo "Already present: test.jpg"
56
  fi
57
 
58
+ # Ask for approval before downloading models and sample files
59
+ echo ""
60
+ echo "This script will download:"
61
+ echo " - Model weights and/or sample files"
62
+ echo ""
63
+ read -p "Continue with downloads? (yes/no): " APPROVAL
64
+ if [[ "${APPROVAL}" != "yes" ]]; then
65
+ echo "Download cancelled by user."
66
+ exit 0
67
+ fi
68
+ echo ""
69
  echo "--- Downloading sample test video ---"
70
  if [[ ! -f test_video.mp4 ]]; then
71
  wget -q -O test_video.mp4 \