Как посмотреть и сохранить значения глубины с помощью SWIFT - PullRequest
Новая
       39

Как посмотреть и сохранить значения глубины с помощью SWIFT

0 голосов
/ 28 января 2020

Я использую некоторую часть кода, упомянутого на слайдах WWD C 2017 (Редактирование изображений с глубиной), чтобы получить глубину и сохранить ее из изображений. https://developer.apple.com/videos/play/wwdc2017/508/. Теперь мне интересно, как я могу получить доступ к значениям глубины и сохранить их в файлах Excel или CSV (я даже не могу смотреть на значения)

Вот мой код

Любая помощь будет так ценится 

   import AppKit
   import AVFoundation
   import CoreImage
   import ImageIO
   import CoreVideo

 let imageURL = URL(fileURLWithPath: "/Users/.../IMG_0876.JPG")
 let source = CGImageSourceCreateWithURL(imageURL as CFURL, nil)
 let sourceProperties = CGImageSourceCopyProperties(source!, nil)
 print("Source properties: \(String(describing: sourceProperties))")

 // Read auxiliary data of type disparity
 let auxDataType = kCGImageAuxiliaryDataTypeDisparity
 let auxDataInfo = CGImageSourceCopyAuxiliaryDataInfoAtIndex(source!, 0, auxDataType)

 // Create a depth data object from auxiliary data
 var depthData = try AVDepthData(fromDictionaryRepresentation: auxDataInfo as! [AnyHashable : Any])
 // Check native depth data type
 if depthData.depthDataType != kCVPixelFormatType_DepthFloat32 {
  // Convert to half-float disparity data
  depthData = depthData.converting(toDepthDataType: kCVPixelFormatType_DepthFloat32)
 }
 // Get the underlying buffer
 let depthMap: CVPixelBuffer = depthData.depthDataMap
 print(depthData)

 /*for i in 1...480 {
     for j in 1...640 {
        print(depthMap)
    }
 }*/
...