如何解决将浮点数列表从文件加载到 MLMultiArray
我有一个包含漂浮的文件时,通过产生的蟒
arr = ... # some np.ndarray of shape (H,W,C)
arr.tofile('floats.b')
我想将其转换为 MLMultiArray
。
我是什么想法是使用以下初始化:
init(dataPointer: UnsafeMutableRawPointer,shape: [NSNumber],dataType: MLMultiArrayDataType,strides: [NSNumber])
但我不知道如何获得 UnsafeMutableRawPointer。
我试图:
- 从
Data
对象获取它。不过好像只想要yield UInt8? - 通过
fopen
获取。还有 UInt8?
var data = UnsafeMutablePointer<Float>.allocate(capacity: 32 * 32 * 220)
fgets(data,Int32(32 * 32 * 220),file) # Compile Error,needs UInt8.
这是怎么做到的?我需要做的是一次将 4 个字节读入 [Float]
。
我能写出以下内容,但我不确定我写对了。
let data = try! Data(contentsOf: url)
var arr = data.withUnsafeBytes { (bytes: UnsafePointer<Float>) -> Array<Float> in
return Array(UnsafeBufferPointer(
start: bytes,count: data.count / MemoryLayout<Float>.size))
}
arr.withUnsafeMutableBufferPointer { ( pointer: inout UnsafeMutableBufferPointer<Float>) in
let raw = UnsafeMutableRawBufferPointer(pointer)
let arr = try! MLMultiArray(dataPointer: raw.baseAddress!,shape: [1,32,220],dataType: MLMultiArrayDataType.float32,strides: [901120,28160,880,4])
}
特别是内存管理似乎不清楚。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。