1 2 3 4 5 6 7 8 9 10 11 12
| import cv2 # pip install opencv-contrib-python-headless
import skimage # pip install scikit-image
im1 = cv2.imread("im1.jpg", cv2.IMREAD_COLOR_BGR)
im2 = cv2.imread("im2.jpg", cv2.IMREAD_COLOR_BGR)
im1 = cv2.cvtColor(im1, cv2.COLOR_BGR2YUV)
im2 = cv2.cvtColor(im2, cv2.COLOR_BGR2YUV)
ssim = (
6 * skimage.metrics.structural_similarity(im1[:, :, 0], im2[:, :, 0], win_size=11, data_range=255, gaussian_weights=True)
+ 1 * skimage.metrics.structural_similarity(im1[:, :, 1], im2[:, :, 1], win_size=11, data_range=255, gaussian_weights=True)
+ 1 * skimage.metrics.structural_similarity(im1[:, :, 2], im2[:, :, 2], win_size=11, data_range=255, gaussian_weights=True)
) / 8 |
Partager