spyrit.misc.metrics

Functions

batch_psnr(torch_batch, output_batch)

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

ssim(I1, I2)

Computes the ssim between two images I1 and I2