matplotlib
lpips
torchmetrics
statistics
