numpy
opencv-python
torch
torchvision
torchaudio
pillow
matplotlib
distinctipy
einops
fast_pytorch_kmeans
joblib==1.2.0
sklearn