numpy
opencv-python
torch
torchvision
torchaudio
pillow
matplotlib
distinctipy
einops
fast_pytorch_kmeans
joblib==1.2.0
scikit-learn==1.0.2