Erich Blume
d51c180fe6
## Summary - Replace abandoned YOLO-NAS-S (320x320, `yolonas`) with YOLOv9-c (640x640, `yolo-generic`) - YOLOv9-c benefits from CUDA Graphs in Frigate 0.17 on the RTX 4080 - Add `export_yolov9` Dagger pipeline and `frigate-export-model` mise task for reproducible model exports - Model already deployed to `sifaka:/volume1/frigate/models/yolov9-c-640.onnx` ## Config changes - `model_type: yolonas` → `yolo-generic` - `input_dtype: int` → `float` - `width/height: 320` → `640` - `path:` → `yolov9-c-640.onnx` ## Deployment and Testing - [ ] Merge and sync Frigate ArgoCD app: `argocd app sync frigate` - [ ] Verify Frigate starts and detects objects at https://nvr.ops.eblu.me - [ ] Confirm GPU inference via Frigate system metrics Reviewed-on: https://forge.ops.eblu.me/eblume/blumeops/pulls/246
55 lines
1.9 KiB
Bash
Executable file
55 lines
1.9 KiB
Bash
Executable file
#!/usr/bin/env bash
|
|
#MISE description="Export YOLOv9 model weights to ONNX for Frigate NVR via Dagger"
|
|
#USAGE flag "--model-size <size>" default="c" help="Model variant: s (small), c (compact), e (extra-large)"
|
|
#USAGE flag "--input-size <pixels>" default="640" help="Input resolution (width=height)"
|
|
#USAGE flag "--deploy" help="Copy exported model to sifaka NAS frigate share"
|
|
|
|
set -euo pipefail
|
|
|
|
MODEL_SIZE="${usage_model_size:-c}"
|
|
INPUT_SIZE="${usage_input_size:-640}"
|
|
DEPLOY="${usage_deploy:-false}"
|
|
OUTPUT_FILE="yolov9-${MODEL_SIZE}-${INPUT_SIZE}.onnx"
|
|
|
|
echo "Exporting YOLOv9-${MODEL_SIZE} (${INPUT_SIZE}x${INPUT_SIZE}) via Dagger..."
|
|
echo ""
|
|
|
|
dagger call export-yolov-9 \
|
|
--model-size="$MODEL_SIZE" \
|
|
--input-size="$INPUT_SIZE" \
|
|
export --path="$OUTPUT_FILE"
|
|
|
|
SIZE=$(du -h "$OUTPUT_FILE" | cut -f1)
|
|
echo ""
|
|
echo "Exported: ${OUTPUT_FILE} (${SIZE})"
|
|
|
|
if [[ "$DEPLOY" == "true" ]]; then
|
|
DEST="sifaka:/volume1/frigate/models/${OUTPUT_FILE}"
|
|
echo "Copying to ${DEST}..."
|
|
scp -O "$OUTPUT_FILE" "$DEST"
|
|
echo "Deployed."
|
|
echo ""
|
|
echo "Update argocd/manifests/frigate/configmap.yaml:"
|
|
echo " model:"
|
|
echo " model_type: yolo-generic"
|
|
echo " width: ${INPUT_SIZE}"
|
|
echo " height: ${INPUT_SIZE}"
|
|
echo " input_tensor: nchw"
|
|
echo " input_dtype: float"
|
|
echo " path: /media/frigate/models/${OUTPUT_FILE}"
|
|
echo " labelmap_path: /labelmap/coco-80.txt"
|
|
else
|
|
echo ""
|
|
echo "To deploy to Frigate NAS:"
|
|
echo " scp ${OUTPUT_FILE} sifaka:/volume1/frigate/models/"
|
|
echo ""
|
|
echo "Then update argocd/manifests/frigate/configmap.yaml:"
|
|
echo " model:"
|
|
echo " model_type: yolo-generic"
|
|
echo " width: ${INPUT_SIZE}"
|
|
echo " height: ${INPUT_SIZE}"
|
|
echo " input_tensor: nchw"
|
|
echo " input_dtype: float"
|
|
echo " path: /media/frigate/models/${OUTPUT_FILE}"
|
|
echo " labelmap_path: /labelmap/coco-80.txt"
|
|
fi
|