MATLAB到Python的代码片段

Question

我有以下MATLAB代码，并想用Python重写它：

mean_ssd = sum(sum((imSrc1 - imSrc2).^2))/numel(imSrc1);

我在Python中所做的工作如下：

width, height = Image.open(open(im1)).size
number_of_pixels = width * height    
mean_ssd = sum[math.pow(sum[(imSrc1 - imSrc2)], 2)] / number_of_pixels

但是我在Python中遇到以下错误：

TypeError: unsupported operand type(s) for -: 'instance' and 'instance'

如何用Python重写MATLAB代码？

谢谢。

Answer 1

试试这个，代替：

import numpy as np

width, height = np.shape(imSrc1)[1], np.shape(imSrc1)[0]
number_of_pixels = width * height    

mean_ssd = ((imSrc1 - imSrc2)**2).sum() / float(number_of_pixels)