(gt, result)
| 123 | |
| 124 | # def get_perceptual_distance(gt, result): |
| 125 | def evaluate_image(gt, result): |
| 126 | gt = np.array(gt) |
| 127 | result = np.array(result) |
| 128 | mse = mean_squared_error(gt / 255, result / 255) |
| 129 | psnr_value = psnr.psnr(gt, result) |
| 130 | ssim_value = ssim.ssim_exact(gt / 255, result / 255) |
| 131 | |
| 132 | gt_tensor = PerceptualSimilarity.util.util.im2tensor(gt[..., :3]) |
| 133 | result_tensor = PerceptualSimilarity.util.util.im2tensor(result[..., :3]) |
| 134 | |
| 135 | init_dm_model() |
| 136 | p_dist = dm_model.forward(gt_tensor, result_tensor) |
| 137 | return mse, ssim_value, psnr_value, p_dist |
| 138 | |
| 139 | |
| 140 | def evaluate_video( |
no test coverage detected