torch
torchvision
numpy
einops
numbers
pillow
opencv-python-headless
argparse
time
matplotlib
warnings
onnx
onnxruntime
scikit-image