SceneKit提取几何深度图渲染全黑问题求助

你的代码渲染出全黑图像，核心存在三个关键问题：

1. 未等待GPU渲染完成就读取纹理 #

调用commandBuffer.commit()后，GPU异步执行渲染命令，此时直接读取深度纹理，拿到的是初始清空值（1.0），而非实际渲染的深度数据。

修复：在读取纹理前添加commandBuffer.waitUntilCompleted()，确保GPU完成渲染操作。

2. 深度纹理数据格式转换错误 #

depth32Float格式的纹理每个像素是4字节的Float值（范围0.0~1.0，0代表最近点，1代表最远点），但你的转换函数直接把这些Float字节当成8位RGB像素处理，数值映射完全错误，导致最终显示全黑。

修复：需要将Float深度值映射到0~255的8位灰度值，再生成图像。

3. 缺少相机视角设置 #

SCNRenderer若未指定相机，会使用默认视角，但可能默认视角未对准场景几何体，导致所有像素都是清空的深度值1.0。

修复：确保场景包含相机节点，或手动创建默认相机并调整位置对准场景。

修正后的完整代码 #

func renderDepthMap(using scene: SCNScene, size: CGSize) -> NSImage? {
    guard let device = MTLCreateSystemDefaultDevice(),
          let commandQueue = device.makeCommandQueue(),
          let commandBuffer = commandQueue.makeCommandBuffer() else {
        return nil
    }

    let renderer = SCNRenderer(device: device, options: nil)
    renderer.autoenablesDefaultLighting = true
    renderer.scene = scene
    
    // 自动添加默认相机（若场景无相机）
    if scene.rootNode.childNodes.first(where: { $0.camera != nil }) == nil {
        let cameraNode = SCNNode()
        cameraNode.camera = SCNCamera()
        cameraNode.position = SCNVector3(x: 0, y: 0, z: 5) // 调整位置对准场景内容
        scene.rootNode.addChildNode(cameraNode)
    }

    // 配置深度纹理
    let textureDescriptor = MTLTextureDescriptor.texture2DDescriptor(pixelFormat: .depth32Float, width: Int(size.width), height: Int(size.height), mipmapped: false)
    textureDescriptor.usage = [.renderTarget, .shaderRead]
    guard let depthTexture = device.makeTexture(descriptor: textureDescriptor) else {
        return nil
    }

    // 配置渲染通道
    let renderPassDescriptor = MTLRenderPassDescriptor()
    renderPassDescriptor.depthAttachment.texture = depthTexture
    renderPassDescriptor.depthAttachment.loadAction = .clear
    renderPassDescriptor.depthAttachment.storeAction = .store
    renderPassDescriptor.depthAttachment.clearDepth = 1.0

    renderer.render(atTime: 0, viewport: CGRect(origin: .zero, size: size), commandBuffer: commandBuffer, passDescriptor: renderPassDescriptor)
    commandBuffer.commit()
    // 等待GPU完成渲染
    commandBuffer.waitUntilCompleted()

    return convertToUIImage(depthTexture: depthTexture)
}

// 修正后的深度纹理转NSImage函数
func convertToUIImage(depthTexture: MTLTexture) -> NSImage? {
    let width = depthTexture.width
    let height = depthTexture.height
    
    // 读取Float格式的原始深度数据
    let floatByteCount = MemoryLayout<Float>.stride * width * height
    guard let floatData = malloc(floatByteCount) else { return nil }
    defer { free(floatData) }
    
    depthTexture.getBytes(floatData, bytesPerRow: MemoryLayout<Float>.stride * width, from: MTLRegionMake2D(0, 0, width, height), mipmapLevel: 0)
    
    // 将Float深度值映射为8位灰度值
    let grayByteCount = width * height
    guard let grayData = malloc(grayByteCount) else { return nil }
    defer { free(grayData) }
    
    let floatBuffer = floatData.bindMemory(to: Float.self, capacity: width * height)
    let grayBuffer = grayData.bindMemory(to: UInt8.self, capacity: width * height)
    
    for i in 0..<(width * height) {
        // 反转深度值：近处显示白色，远处显示黑色（可根据需求移除1.0-）
        let normalizedDepth = 1.0 - floatBuffer[i]
        grayBuffer[i] = UInt8(normalizedDepth * 255.0)
    }
    
    // 创建单通道灰度CGImage
    let colorSpace = CGColorSpaceCreateDeviceGray()
    let bitmapInfo = CGBitmapInfo(rawValue: CGImageAlphaInfo.none.rawValue)
    guard let context = CGContext(data: grayData,
                                  width: width,
                                  height: height,
                                  bitsPerComponent: 8,
                                  bytesPerRow: width,
                                  space: colorSpace,
                                  bitmapInfo: bitmapInfo.rawValue),
          let cgImage = context.makeImage() else {
        return nil
    }

    let imageSize = NSSize(width: width, height: height)
    return NSImage(cgImage: cgImage, size: imageSize)
}

关键修正说明 #

• 新增commandBuffer.waitUntilCompleted()确保渲染完成后再读取纹理数据。
• 转换函数先读取Float格式深度值，再映射为8位灰度值，解决格式不匹配问题。
• 添加自动创建相机逻辑，避免因无相机导致全远距深度值。
• 可选反转深度值，让近处物体显示白色，更符合视觉预期（不需要可移除1.0 -）。

内容的提问来源于stack exchange，提问作者Mane Manero