Exporting fp16 onnx models.

2024-11-29 17:37:44 +01:00
parent 7b9505ca02
commit 19015c6326
10 changed files with 107 additions and 13 deletions
@@ -28,5 +28,8 @@ RUN echo 'export LD_LIBRARY_PATH=/mmdeploy/../mmdeploy-dep/onnxruntime-linux-x64
 # Show images
 RUN apt-get update && apt-get install -y --no-install-recommends python3-tk

+# Tool for fp16 conversion
+RUN pip3 install --upgrade --no-cache-dir onnxconverter_common 
+
 WORKDIR /mmdeploy/
 CMD ["/bin/bash"]