Search code examples
iosswiftcoreml

VNCoreMLFeatureValueObservation VS class VNClassificationObservation


I was researching on CoreML to create Machine learning app. I was reading the apple Documentation and found this two class VNCoreMLFeatureValueObservation and VNClassificationObservation. After reading documentation i was confused on what kind of model should i use these class. Also apple documentation provide different model such as mobileNet, SqueezeNet, Places205-GoogLeNet, ResNet50 and VGG16


Solution

  • The VNClassificationObservation is returned when the model is a classifier. It now outputs a dictionary of [String: Double] that has the probability score for every class in the model. The models you listed are all classifiers.

    The VNCoreMLFeatureValueObservation is returned when the model is not a classifier. Instead of a dictionary, such models usually output an MLMultiArray object. You willhave to do your own post-processing to interpret the data from this kind of output.