add midas model building to toolkit

toolkit/Dockerfile

@@ -8,6 +8,8 @@ RUN apt-get update && \
     apt-get install -y --no-install-recommends \
       wget \
       git \
+      git-lfs \
+      ca-certificates \
       libgl1-mesa-glx && \
     rm -rf /var/lib/apt/lists/*
 
@@ -89,10 +91,15 @@ RUN pip install --no-cache-dir \
     pyyaml \
     "tensorflow<=2.16.0rc0"
 
+# fixes for python package conflicts, we are forcing specific versions.
+RUN pip install --no-cache-dir \
+    "ml-dtypes==0.5.4" "onnx==1.18.0" "onnxruntime==1.18.1"
+
 # download other onnx models
 RUN git clone --depth 1 https://github.com/swdee/go-rknnlite-build.git /opt/go-rknnlite-build && \
-    git -C /opt/go-rknnlite-build fetch --depth 1 origin ce8b5ce1dc53b1c38324e7506374731ad21070c8 && \
-    git -C /opt/go-rknnlite-build checkout FETCH_HEAD
+    git -C /opt/go-rknnlite-build fetch --depth 1 origin 8380b2265b4becf2dda4c7ff52b9d57f8c718f09 && \
+    git -C /opt/go-rknnlite-build checkout FETCH_HEAD && \
+    git -C /opt/go-rknnlite-build lfs pull
 
 # copy our modified mobilenet.py script into the rknn_model_zoo directory
 RUN cp /opt/go-rknnlite-build/mobilenet-batch/mobilenet-rknn.py /opt/rknn_model_zoo/examples/mobilenet/python/mobilenet-rknn-batch.py

toolkit/compile-models.sh

@@ -32,6 +32,7 @@ MODELS=(
   "yolov8            convert.py        /opt/go-rknnlite-build/yolonas-s.onnx        i8    ''    yolonas-s"
   "mobilenet         mobilenet-rknn-batch.py      ../model/mobilenetv2-12.onnx      i8    --model     mobilenetv2-batch8"
   "osnet-market1501  build|onnx_to_rknn.py        osnet_x1_0_market_256x128.onnx    i8    ''          osnet-market1501-batch8"
+  "midas             build|onnx_to_rknn.py        dpt_swin2_tiny_256.onnx           i8    ''          dpt_swin2_tiny_256"
 )
 
 # compile all entries (or just filter) for one platform