numpy
opencv-python
torch
torchvision
torchaudio
pillow
matplotlib
distinctipy
einops
fast_pytorch_kmeans