add cli

machine-learning/export/ann/Dockerfile

@@ -1,28 +1,35 @@
 FROM mambaorg/micromamba:bookworm-slim@sha256:333f7598ff2c2400fb10bfe057709c68b7daab5d847143af85abcf224a07271a as builder
 
-ENV TRANSFORMERS_CACHE=/cache \
-  PYTHONDONTWRITEBYTECODE=1 \
-  PYTHONUNBUFFERED=1 \
-  PATH="/opt/venv/bin:$PATH"
-
-WORKDIR /export/ann
-
 USER root
 RUN apt-get update && apt-get install -y --no-install-recommends \
   build-essential \
+  cmake \
   curl \
   git
-
 USER $MAMBA_USER
-COPY --chown=$MAMBA_USER:$MAMBA_USER env.yaml ./
-RUN micromamba install -y -f env.yaml
-COPY --chown=$MAMBA_USER:$MAMBA_USER *.sh *.cpp ./
 
-ENV ARMNN_PATH=/export/ann/armnn
+WORKDIR /home/mambauser
+ENV ARMNN_PATH=armnn
+COPY --chown=$MAMBA_USER:$MAMBA_USER scripts/* .
 RUN ./download-armnn.sh && \
-    ./build-converter.sh && \
-    ./build.sh
-COPY --chown=$MAMBA_USER:$MAMBA_USER run.py ./
+  ./build-converter.sh && \
+  ./build.sh
 
-ENTRYPOINT ["/usr/local/bin/_entrypoint.sh"]
-CMD ["python", "run.py"]
+COPY --chown=$MAMBA_USER:$MAMBA_USER conda-lock.yml .
+RUN micromamba create -y -p /home/mambauser/venv -f conda-lock.yml && \
+  micromamba clean --all --yes
+ENV PATH="/home/mambauser/venv/bin:${PATH}"
+
+FROM gcr.io/distroless/base-debian12
+# FROM mambaorg/micromamba:bookworm-slim@sha256:333f7598ff2c2400fb10bfe057709c68b7daab5d847143af85abcf224a07271a
+
+WORKDIR /export/ann
+ENV PYTHONDONTWRITEBYTECODE=1 \
+  LD_LIBRARY_PATH=/export/ann/armnn \
+  PATH="/opt/venv/bin:${PATH}"
+
+COPY --from=builder /home/mambauser/armnnconverter /home/mambauser/armnn ./
+COPY --from=builder /home/mambauser/venv /opt/venv
+COPY --chown=$MAMBA_USER:$MAMBA_USER onnx2ann onnx2ann
+
+ENTRYPOINT ["python", "-m", "onnx2ann"]