PetsClassifier Demo App

A multi-platform app for showcasing my Pets Image Classifier CoreML Model

Usage:

// create the model instance
let model = try PetsClassifier()

// prepare image
let uiImage: UIImage = .....
let nsImage: NSImage = .....

if let buffer = uiImage.colorPixelBuffer() { // or nsImage.colorPixelBuffer() 
    let input = PetsClassifierInput(image: buffer)
    let output = try await model.prediction(input: input)

    print(output.target) // cat
    print(output.targetProbability) // { "cat": 0.99, "dog": 0.1, "rabbit": 0.01 }
}

Image Conversion Method's Credits:

• UIImage -> CVPixelBuffer francoismarceau29
• NSImage -> CVPixelBuffer DennisWeidmann