1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
FROM ubuntu:24.04
RUN apt-get update && apt-get install -y python3 python3-venv cmake build-essential git python3-pip \
python3-dev libgl1-mesa-dev libglu1-mesa-dev freeglut3-dev\
libglib2.0-0 \
&& rm -rf /var/lib/apt/lists/*
RUN adduser appuser
RUN mkdir /yolox && chown -R appuser /yolox
USER appuser
RUN git clone https://github.com/Megvii-BaseDetection/YOLOX.git /yolox\
&& cd /yolox \
&& git reset --hard 6ddff4824372906469a7fae2dc3206c7aa4bbaee
# https://github.com/Megvii-BaseDetection/YOLOX/pull/1826
RUN sed -i -e 's/onnx-simplifier==0.4.10/onnxsim==0.6.2/g' /yolox/requirements.txt
RUN python3 -m venv /yolox/venv
ARG PYTHON_BIN=/yolox/venv/bin/python3
RUN ${PYTHON_BIN} --version
RUN cd /yolox && ${PYTHON_BIN} -m pip install -r requirements.txt
RUN cd /yolox && ${PYTHON_BIN} -m pip install onnxscript==0.6.2
RUN cd /yolox && ${PYTHON_BIN} -m pip install onnxruntime==1.24.4
RUN sed -i -e 's/torch.onnx._export/torch.onnx.export/g' /yolox/tools/export_onnx.py
RUN cd /yolox && ${PYTHON_BIN} setup.py install
CMD ${PYTHON_BIN} /yolox/tools/export_onnx.py --output-name /data/yolox_tiny.onnx -n yolox-tiny -c /data/yolox_tiny.pth -o 11