如何调整 MLMultiArray 的大小以适合我的相机纹理大小？

Question

我正在使用 DeepLabV3 mlmodel 对人进行一些语义分割。预测后的输出是 513*513 的 MLMultiArray。目前我正在将我的相机输出调整到这个大小以应用分段数组。

如何调整 MLMultiArray 的大小以匹配我的相机纹理大小？

if let observations = request.results as? [VNCoreMLFeatureValueObservation],
        let segmentationmap = observations.first?.featureValue.multiArrayValue {
        
        // row - 513, col - 513
        guard let row = segmentationmap.shape[0] as? Int,
            let col = segmentationmap.shape[1] as? Int else {
                return
        }
}

Answer 1

下面是一些在 iOS 应用程序中使用 DeepLab V3 的演示代码：https://github.com/hollance/SemanticSegmentationMetalDemo