spyrit.misc.metrics

Functions

batch_psnr(torch_batch, output_batch)

batch_psnr_(torch_batch, output_batch[, r])

batch_psnr_vid(input_batch, output_batch)

batch_ssim(torch_batch, output_batch)

batch_ssim_vid(input_batch, output_batch)

compare_nets_unsupervised(net_list, ...)

compare_video_nets_supervised(net_list, ...)

dataset_meas(dataloader, model, device)

dataset_psnr(dataloader, model, device)

dataset_psnr_ssim(dataloader, model, device)

dataset_psnr_ssim_fcl(dataloader, model, device)

dataset_ssim(dataloader, model, device)

print_mean_std(x[, tag])

psnr(I1, I2)

Computes the psnr between two images I1 and I2

psnr_(img1, img2[, r])

Computes the psnr between two image with values expected in a given range

psnr_torch(img_gt, img_rec[, mask, dim, img_dyn])

Computes the Peak Signal-to-Noise Ratio (PSNR) between two images.

ssim(I1, I2)

Computes the ssim between two images I1 and I2