mirror of
https://github.com/blakeblackshear/frigate.git
synced 2024-12-29 00:06:19 +01:00
21 lines
624 B
Plaintext
21 lines
624 B
Plaintext
|
#!/command/with-contenv bash
|
||
|
|
# shellcheck shell=bash
|
||
|
|
# Compile YoloV8 ONNX files into ROCm MIGraphX files
|
||
|
|
|
||
|
|
OVERRIDE=$(cd /opt/frigate && python3 -c 'import frigate.detectors.plugins.rocm as rocm; print(rocm.auto_override_gfx_version())')
|
||
|
|
|
||
|
|
if ! test -z "$OVERRIDE"; then
|
||
|
|
echo "Using HSA_OVERRIDE_GFX_VERSION=${OVERRIDE}"
|
||
|
|
export HSA_OVERRIDE_GFX_VERSION=$OVERRIDE
|
||
|
|
fi
|
||
|
|
|
||
|
|
for onnx in /config/model_cache/yolov8/*.onnx
|
||
|
|
do
|
||
|
|
mxr="${onnx%.onnx}.mxr"
|
||
|
|
if ! test -f $mxr; then
|
||
|
|
echo "processing $onnx into $mxr"
|
||
|
|
/opt/rocm/bin/migraphx-driver compile $onnx --optimize --gpu --enable-offload-copy --binary -o $mxr
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
|