CVPixelBuffer转换为UIImage时出现内存泄漏导致App偶发崩溃怎么解决？

• 你当前使用的VTCreateCGImageFromCVPixelBuffer接口生成的CGImage会隐式持有原始CVPixelBuffer的内存引用，相机采集属于高频场景（每秒30-60帧），如果没有及时释放相关对象，内存会快速堆积触发OOM崩溃
• 原实现没有处理CVPixelBuffer的内存锁逻辑，可能出现读写冲突导致的异常崩溃
• 高频转换场景下如果没有包裹自动释放池，自动释放对象会等到当前RunLoop结束才释放，短时间内存峰值过高

方案1：使用CoreGraphics手动绘制实现转换（稳定性最高） #

替换原有VideoToolbox接口，手动处理像素锁和上下文绘制，内存释放逻辑更可控：

extension UIImage {
    public convenience init?(pixelBuffer: CVPixelBuffer) {
        // 锁定像素缓冲区内存
        CVPixelBufferLockBaseAddress(pixelBuffer, .readOnly)
        defer {
            // 退出时自动解锁
            CVPixelBufferUnlockBaseAddress(pixelBuffer, .readOnly)
        }
        
        let baseAddress = CVPixelBufferGetBaseAddress(pixelBuffer)
        let width = CVPixelBufferGetWidth(pixelBuffer)
        let height = CVPixelBufferGetHeight(pixelBuffer)
        let bytesPerRow = CVPixelBufferGetBytesPerRow(pixelBuffer)
        let colorSpace = CGColorSpaceCreateDeviceRGB()
        
        // 根据像素缓冲区格式创建位图上下文，这里默认是BGRA格式（相机采集常用格式）
        guard let context = CGContext(
            data: baseAddress,
            width: width,
            height: height,
            bitsPerComponent: 8,
            bytesPerRow: bytesPerRow,
            space: colorSpace,
            bitmapInfo: CGImageAlphaInfo.premultipliedFirst.rawValue | CGBitmapInfo.byteOrder32Little.rawValue
        ) else {
            return nil
        }
        
        guard let cgImage = context.makeImage() else {
            return nil
        }
        
        self.init(cgImage: cgImage)
    }
}

方案2：原有实现基础上添加内存管控 #

如果要保留VTCreateCGImageFromCVPixelBuffer的实现，添加自动释放池包裹转换逻辑，确保临时对象及时释放：

extension UIImage {
    public convenience init?(pixelBuffer: CVPixelBuffer) {
        var cgImage: CGImage?
        // 包裹自动释放池
        autoreleasepool {
            VTCreateCGImageFromCVPixelBuffer(pixelBuffer, options: nil, imageOut: &cgImage)
        }
        guard let cgImage = cgImage else {
            return nil
        }
        self.init(cgImage: cgImage)
    }
}

• 注意：调用转换逻辑的外层如果是在相机采集回调中，也需要额外加一层autoreleasepool包裹所有处理逻辑，避免帧对象堆积

额外优化建议 #

• 非必要场景不要长期持有转换后的UIImage对象，处理完业务逻辑（比如AI检测、帧编码）后直接释放
• 如果只是做预览渲染，不要转UIImage，直接用AVSampleBufferDisplayLayer或者Metal渲染原始CVPixelBuffer，可以节省90%以上的转换内存开销
• 控制相机采集帧率，不需要高帧率的场景将采集帧率降到30fps以下

内容的提问来源于stack exchange，提问作者Federico Rotoli