Benchmark Core ML model in iOS.

mlSpeedTests.swift

import CoreML
import XCTest
@testable import mlsample

class mlsampleTests: XCTestCase {
    
    override func setUp() {
        super.setUp()
    }
    
    override func tearDown() {
        super.tearDown()
    }
    
    func testSpeed() {
        guard let input_data = try? MLMultiArray(shape:[3, 160, 160], dataType:MLMultiArrayDataType.double) else {
            fatalError("Unexpected runtime error. MLMultiArray")
        }
        let model = mobile_unet_160_100_100()
        // It's for excluding 1st measurement.
        do {
            try model.prediction(data: input_data)
            print("predicted------------")
        } catch  {
            print("error------------")
        }
        print("--------started------------")
        self.measure() {
            try! model.prediction(data: input_data)
        }
    }
}