torch
torchmetrics
scikit-image
Pillow
