MLMultiArray to Double Array

README.md

Sample Data

Get cat.bin

Xcode Resources Add cat.bin

Playground

import CoreML

let url = Bundle.main.url(forResource: "cat", withExtension: "bin")
let data = try! Data(contentsOf: url!)
let ptr = UnsafeMutableRawPointer(mutating: (data as NSData).bytes)
let m = try? MLMultiArray(dataPointer: ptr,
                         shape: [3, 360, 480],
                         dataType: .double,
                         strides: [NSNumber(value: 360*480), 480, 1])

let mm = m!
let length = mm.count
let doublePtr =  mm.dataPointer.bindMemory(to: Double.self, capacity: length)
let doubleBuffer = UnsafeBufferPointer(start: doublePtr, count: length)
let output = Array(doubleBuffer)

resources.png