torch == 1.5.0
torchvision == 0.6.0
onnx == 1.8.1
numpy==1.19.5
Pillow == 8.2.0
opencv-python == 4.5.2.54
yacs == 0.1.8
pytorch-ignite == 0.4.5
timm=0.4.12
protobuf=3.20.1
tqdm
onnx-simplifier