[Solved] ValueError: cannot reshape array of size 509760 into shape (500,353,3)

ValueError: cannot reshape array of size 509760 into shape (500,353,3) appears because of image conversion problems

Write a conversion function:

def reshape_cv(img):
    # resize图片大小 先将原本的(224,222,3) ---> (28,28,3)
    pred_img = cv.resize(img, (500, 353))
    # 转换np数组格式
    pred_img = np.array(pred_img)
    # 重新reshape图片
    pred_img = pred_img.reshape(500, 353, 3)
    # 查看reshape后的图片shape
    print(pred_img.shape)
    return pred_img

Evaluate ssim and psnr:

import cv2 as cv
import numpy as np
from skimage.metrics import structural_similarity as compare_ssim
from skimage.metrics import peak_signal_noise_ratio as compare_psnr
from skimage.metrics import mean_squared_error as compare_mse


def reshape_cv(img):
    # resize图片大小 先将原本的(224,222,3) ---> (28,28,3)
    pred_img = cv.resize(img, (500, 353))
    # 转换np数组格式
    pred_img = np.array(pred_img)
    # 重新reshape图片
    pred_img = pred_img.reshape(500, 353, 3)
    # 查看reshape后的图片shape
    print(pred_img.shape)
    return pred_img

img1 = cv.imread("./img/111_before.jpg")
print(img1.shape)
img2 = cv.imread("./img/222_before.jpg")
print(img2.shape)
img2 = reshape_cv(img2)
print(img2.shape)

ssim = compare_ssim(img1, img2, win_size=11, data_range=255, multichannel=True)
print(ssim)