我正在尝试使用 Apple 上给定的 inceptionV3 模型在我的项目中结合 CoreML 和 ARKit website .

我从 ARKit (Xcode 9 beta 3) 的标准模板开始

我没有实例化一个新的相机 session ，而是重用了由 ARSCNView 启动的 session 。

在我的 viewDelegate 的最后，我写:

sceneView.session.delegate = self

然后我扩展我的 viewController 以符合 ARSessionDelegate 协议(protocol)(可选协议(protocol))

// MARK: ARSessionDelegate
extension ViewController: ARSessionDelegate {

    func session(_ session: ARSession, didUpdate frame: ARFrame) {

        do {
            let prediction = try self.model.prediction(image: frame.capturedImage)
            DispatchQueue.main.async {
                if let prob = prediction.classLabelProbs[prediction.classLabel] {
                    self.textLabel.text = "\(prediction.classLabel) \(String(describing: prob))"
                }
            }
        }
        catch let error as NSError {
            print("Unexpected error ocurred: \(error.localizedDescription).")
        }
    }
}

起初我尝试了该代码，但后来注意到 inception 需要一个 Image 类型的像素缓冲区。 <><299,299>。

虽然没有重新开始，但我想我会调整框架的大小，然后尝试从中获得预测。我正在使用此功能调整大小(取自 https://github.com/yulingtianxia/Core-ML-Sample )

func resize(pixelBuffer: CVPixelBuffer) -> CVPixelBuffer? {
    let imageSide = 299
    var ciImage = CIImage(cvPixelBuffer: pixelBuffer, options: nil)
    let transform = CGAffineTransform(scaleX: CGFloat(imageSide) / CGFloat(CVPixelBufferGetWidth(pixelBuffer)), y: CGFloat(imageSide) / CGFloat(CVPixelBufferGetHeight(pixelBuffer)))
    ciImage = ciImage.transformed(by: transform).cropped(to: CGRect(x: 0, y: 0, width: imageSide, height: imageSide))
    let ciContext = CIContext()
    var resizeBuffer: CVPixelBuffer?
    CVPixelBufferCreate(kCFAllocatorDefault, imageSide, imageSide, CVPixelBufferGetPixelFormatType(pixelBuffer), nil, &resizeBuffer)
    ciContext.render(ciImage, to: resizeBuffer!)
    return resizeBuffer
} 

不幸的是，这还不足以让它发挥作用。这是捕获的错误:

Unexpected error ocurred: Input image feature image does not match model description.
2017-07-20 AR+MLPhotoDuplicatePrediction[928:298214] [core] 
    Error Domain=com.apple.CoreML Code=1 
    "Input image feature image does not match model description" 
    UserInfo={NSLocalizedDescription=Input image feature image does not match model description, 
    NSUnderlyingError=0x1c4a49fc0 {Error Domain=com.apple.CoreML Code=1 
    "Image is not expected type 32-BGRA or 32-ARGB, instead is Unsupported (875704422)" 
    UserInfo={NSLocalizedDescription=Image is not expected type 32-BGRA or 32-ARGB, instead is Unsupported (875704422)}}}

不确定我能从这里做什么。

如果有更好的建议将两者结合起来，我会洗耳恭听。

编辑:我也试过resizePixelBuffer @dfd 建议的 YOLO-CoreML-MPSNNGraph 方法，错误完全相同。

Edit2:所以我将像素格式更改为 kCVPixelFormatType_32BGRA(与 resizePixelBuffer 中传递的 pixelBuffer 的格式不同)。

let pixelFormat = kCVPixelFormatType_32BGRA // line 48

我不再有错误了。但是，一旦我尝试做出预测，AVCaptureSession 就会停止。似乎我遇到了同样的问题 Enric_SA 在 apple developers forum 上运行.

Edit3:所以我尝试实现 rickster 解决方案。与 inceptionV3 配合良好。我想尝试一个特征观察(VNClassificationObservation)。此时，它无法使用 TinyYolo .边界是错误的。想弄明白。

最佳答案

不要自己处理图像以将它们提供给 Core ML。使用 Vision . (不，不是那个。This one。)Vision 采用 ML 模型和多种图像类型中的任何一种 (including CVPixelBuffer)，并自动将图像设置为正确的尺寸、纵横比和像素格式，供模型评估，然后给你模型的结果。

这是您需要的代码的粗略框架:

var request: VNRequest

func setup() {
    let model = try VNCoreMLModel(for: MyCoreMLGeneratedModelClass().model)
    request = VNCoreMLRequest(model: model, completionHandler: myResultsMethod)
}

func classifyARFrame() {
    let handler = VNImageRequestHandler(cvPixelBuffer: session.currentFrame.capturedImage,
        orientation: .up) // fix based on your UI orientation
    handler.perform([request])
}

func myResultsMethod(request: VNRequest, error: Error?) {
    guard let results = request.results as? [VNClassificationObservation]
        else { fatalError("huh") }
    for classification in results {
        print(classification.identifier, // the scene label
              classification.confidence)
    }
}

参见 this answer到另一个问题以获得更多指示。

关于swift - 结合 CoreML 和 ARKit，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/45221746/