six
numpy
scipy
matplotlib
pillow
colorama
progiter
protobuf
opencv-python
torch>=1.5
torchvision>=0.6
tensorboard
onnx
packaging
fast-histogram