diff --git a/docker/rockchip/rk.mk b/docker/rockchip/rk.mk index 81cdb4ed8..c8278f68b 100644 --- a/docker/rockchip/rk.mk +++ b/docker/rockchip/rk.mk @@ -9,7 +9,7 @@ build-rk: version docker buildx bake --file=docker/rockchip/rk.hcl rk \ --set rk.tags=$(IMAGE_REPO):${GITHUB_REF_NAME}-$(COMMIT_HASH)-rk -push-rk: version +push-rk: build-rk docker buildx bake --file=docker/rockchip/rk.hcl rk \ - --set rk.tags=crzynik/frigate:rk \ + --set rk.tags=$(IMAGE_REPO):${GITHUB_REF_NAME}-$(COMMIT_HASH)-rk \ --push \ No newline at end of file diff --git a/frigate/util/rknn_converter.py b/frigate/util/rknn_converter.py index 31493cf55..831477b58 100644 --- a/frigate/util/rknn_converter.py +++ b/frigate/util/rknn_converter.py @@ -182,7 +182,16 @@ def convert_onnx_to_rknn( rknn = RKNN(verbose=True) rknn.config(**config) - if rknn.load_onnx(model=onnx_path) != 0: + if model_type == "jina-clip-v1-vision": + load_output = rknn.load_onnx( + model=onnx_path, + inputs=["pixel_values"], + input_size_list=[[1, 3, 224, 224]], + ) + else: + load_output = rknn.load_onnx(model=onnx_path) + + if load_output != 0: logger.error("Failed to load ONNX model") return False