scikit-image
matplotlib
torch
torchvision
torchaudio
torchinfo
