Running this code base in docker

[VeeranjaneyuluToka]

Hi, anybody recently run this codebase in docker, if so, could you please share the details here. That would be great help to me to train a model with my own dataset.

Thanks

[akonanykhin]

I've used the following package versions in python:3.10 docker image:

torch==2.4.1
torchvision
opencv-python-headless==4.11.0.86
tensorflow==2.15.1
tensorboard==2.15.2
openvino-dev==2024.6.0
nncf==2.15.0
onnx==1.17.0
onnxruntime==1.20.1
onnx-simplifier==0.4.36
pycocotools==2.0.8
pyyaml
loguru
tqdm
thop
ninja
tabulate
psutil
onnx2tf==1.26.8
tf-keras==2.15.1
onnx_graphsurgeon==0.5.5
sng4onnx==1.0.4

Some packages are not needed for the repo per se, like onnx2tf.
Unpinned versions are not important.