Swift中如何判断两个OpenCV Mat是否相等及排查代码异常

一、Swift中测试两个OpenCV Mat是否相等的正确方式 #

不要用isSameMat或isEqual，这类方法判断的是Mat对象的内存同一性，而非元素内容相等。正确测试方式有两种：

方式1：高效的OpenCV原生方法 #

利用OpenCV的compare生成差异矩阵，再统计非零元素数量：

func areMatsEqual(_ mat1: Mat, _ mat2: Mat) -> Bool {
    // 先检查尺寸、类型是否一致，不一致直接返回不等
    guard mat1.size() == mat2.size() && mat1.type() == mat2.type() else {
        return false
    }
    let diff = Mat()
    // 生成差异矩阵：元素不等时标记为非零
    cv.compare(mat1, mat2, diff, CMP_NE)
    // 非零元素为0则说明所有元素相等
    return cv.countNonZero(diff) == 0
}

方式2：逐元素对比（适合调试） #

注意要提取get方法返回数组中的实际元素值（单通道Mat返回长度为1的数组）：

func areMatsEqualByElement(_ mat1: Mat, _ mat2: Mat) -> Bool {
    guard mat1.size() == mat2.size() && mat1.type() == mat2.type() else {
        return false
    }
    for row in 0 ..< mat1.rows() {
        for col in 0 ..< mat1.cols() {
            let val1 = mat1.get(row: row, col: col)[0]
            let val2 = mat2.get(row: row, col: col)[0]
            if val1 != val2 {
                return false
            }
        }
    }
    return true
}

二、Stack Overflow矩阵相等判断逻辑的Swift迁移 #

Stack Overflow上判断OpenCV矩阵相等的核心逻辑是：

1. 先验证矩阵的尺寸、数据类型是否完全一致
2. 若一致，计算两矩阵的差异，确认无差异元素

对应到Swift中，就是上述的areMatsEqual函数：

• 第一步的尺寸、类型检查直接通过guard语句实现
• 第二步用cv.compare生成差异矩阵，cv.countNonZero统计差异元素数量，完全贴合原逻辑

三、你的代码异常原因与修复 #

异常根源 #

1. isSameMat的误用：isSameMat仅判断两个Mat是否指向同一内存块（即是否为同一个对象实例），你创建了两个独立的Mat对象，即便元素完全相同，isSameMat也会返回false，导致断言触发。
2. 逐元素对比的疏漏：你直接对比get方法返回的数组对象，而非数组内的实际元素值，导致即使元素存在差异也无法检测到。

修复后的完整代码 #

let output = Mat(rows: input.rows(), cols: input.cols(), type: CvType.CV_8UC1)
let test = Mat(rows: input.rows(), cols: input.cols(), type: CvType.CV_8UC1)
// ... fill test and output ...

func areMatsEqual(_ mat1: Mat, _ mat2: Mat) -> Bool {
    guard mat1.size() == mat2.size() && mat1.type() == mat2.type() else {
        return false
    }
    let diff = Mat()
    cv.compare(mat1, mat2, diff, CMP_NE)
    return cv.countNonZero(diff) == 0
}

if !areMatsEqual(test, output) {
    // 调试用逐元素检查：提取数组中的元素值对比
    for x in 0 ..< output.rows() {
        for y in 0 ..< output.cols() {
            let value = output.get(row: x, col: y)[0]
            let t = test.get(row: x, col: y)[0]
            if t != value {
                print("first mismatch row \(x) cols \(y)")
                break
            }
        }
    }
    assert(areMatsEqual(test, output), "Mats are not equal")
}

内容的提问来源于stack exchange，提问作者Anton Tropashko