svm.py 文件源码

def mean_squares(self):
        self.threshold = 0.00000000001        
        b_i,g_i,r_i = cv2.split(self.image_i)
        b_j,g_j,r_j = cv2.split(self.image_j)

        error_b = b_i - b_j
        error_g = g_i - g_j
        error_r = r_i - r_j

        error_b = error_b.flatten()
        error_g = error_g.flatten()
        error_r = error_r.flatten()

        mse_b = float(numpy.dot(error_b,error_b))/len(error_b) 
        mse_g = float(numpy.dot(error_g,error_g))/len(error_g)

        mse_r = float(numpy.dot(error_r,error_r))/len(error_r)
        self.measure = (mse_b + mse_g + mse_r)/3
        self.assertLess(self.measure, self.threshold)