cristhian aguilera
fd6993e294
- Add OpenVocab engine: YOLO-World ONNX detection with runtime vocabulary via CLIP text embeddings (two-input model support) - Add header-only C++ CLIP BPE tokenizer (no external deps) - Add detect(cv::Mat) overload to OpenVocab for camera-free testing - Add standalone test binary (test_open_vocab) using test/assets images - Add export scripts for YOLO-World and CLIP text encoder (uv/PEP 723) - Add installation scripts for Linux x64, ARM (Jetson), and Windows - Update CMakeLists to use find_package(onnxruntime) instead of FetchContent - Update README and AGENTS with env vars, setup workflow, and test usage
48 lines
1.8 KiB
PowerShell
Executable File
48 lines
1.8 KiB
PowerShell
Executable File
# Install build dependencies for dora-zed on Windows x64.
|
|
# Skips any component that is already installed.
|
|
# Run from an elevated PowerShell prompt.
|
|
#
|
|
# Requirements before running:
|
|
# - CUDA Toolkit 12.x: https://developer.nvidia.com/cuda-downloads
|
|
# - CMake 3.21+: https://cmake.org/download/
|
|
# - Rust toolchain: https://rustup.rs
|
|
|
|
$ErrorActionPreference = "Stop"
|
|
|
|
$ZED_INSTALL_DIR = "C:\Program Files (x86)\ZED SDK"
|
|
$ORT_VERSION = "1.24.4"
|
|
$ORT_NAME = "onnxruntime-win-x64-gpu_cuda13-$ORT_VERSION"
|
|
$ORT_URL = "https://github.com/microsoft/onnxruntime/releases/download/v$ORT_VERSION/$ORT_NAME.zip"
|
|
$ORT_DEST = "C:\onnxruntime"
|
|
|
|
# --- ZED SDK ---
|
|
if (Test-Path $ZED_INSTALL_DIR) {
|
|
Write-Host ">>> ZED SDK already installed at '$ZED_INSTALL_DIR' — skipping."
|
|
} else {
|
|
Write-Host ">>> ZED SDK: download the installer from the release page and run it."
|
|
Write-Host " https://www.stereolabs.com/developers/release"
|
|
Write-Host " After install, ZED SDK will be at '$ZED_INSTALL_DIR'."
|
|
Write-Host ""
|
|
}
|
|
|
|
# --- ONNX Runtime ---
|
|
if (Test-Path "$ORT_DEST\lib\onnxruntime.dll") {
|
|
Write-Host ">>> ONNX Runtime already installed at '$ORT_DEST' — skipping."
|
|
} else {
|
|
Write-Host ">>> Downloading ONNX Runtime $ORT_VERSION GPU (CUDA 13)..."
|
|
$tmp = "$env:TEMP\$ORT_NAME.zip"
|
|
Invoke-WebRequest -Uri $ORT_URL -OutFile $tmp
|
|
|
|
Write-Host ">>> Extracting to $ORT_DEST..."
|
|
Expand-Archive -Path $tmp -DestinationPath $env:TEMP -Force
|
|
if (Test-Path $ORT_DEST) { Remove-Item $ORT_DEST -Recurse -Force }
|
|
Move-Item "$env:TEMP\$ORT_NAME" $ORT_DEST
|
|
Remove-Item $tmp
|
|
|
|
Write-Host ">>> ONNX Runtime installed to $ORT_DEST"
|
|
Write-Host " Pass it to CMake: cmake .. -DCMAKE_PREFIX_PATH=C:\onnxruntime"
|
|
}
|
|
|
|
Write-Host ""
|
|
Write-Host ">>> Done."
|