PodTFRecords 0.8.0

PodTFRecords 0.8.0

×

由 Hai Feng Kao 维护。

 
依赖库
SwiftProtobuf> 1.6.0
CryptoSwift> 1.0.0
 

PodTFRecords 0.8.0

  • 由
  • Hai\ Feng\ Kao

SwiftTFRecords

Swift的TFRecords (.tfrecord) 文件格式构造器和读取器

TFRecords格式简要介绍这里，并被描述为向TensorFlow输入数据的推荐格式这里和这里。

该库便利地在Swift中直接生成和导入TFRecords格式的数据。

该库不是“官方”的 - 它不是TensorFlow的一部分，也不是由TensorFlow团队维护的。

用法 - 生成一个TFRecords文件

下面的示例涵盖了生成TFRecords文件的推荐API使用方法。

import SwiftTFRecords

var record = Record()

record["Int"] = 1
record["Float"] = 2.3
record["String"] = "Jacopo 😃"
record["FloatArray"] = [2.1, 2.2, 2.3]
record["Bytes"] = Feature.Bytes(Data([1, 2, 3, 4])
record["IntArray"] = Feature.IntArray([1, 2, 3, 4])

let tfRecord = TFRecords(withRecords: [record])

tfRecord.data.write(to: URL(fileURLWithPath: "file.tfrecord"))

用法 - 导入一个TFRecords文件

以下示例涵盖了读取TFRecords文件的推荐API使用方法。

import SwiftTFRecords

let data = try Data(contentsOf: URL(fileURLWithPath: "file.tfrecord"))

let tfRecord = TFRecords(withData: data)

for record in tfRecord.records {
    print("---")
    print(record["Int"]?.toInt())
    print(record["Float"]?.toFloat())
    print(record["String"]?.toString())
    print(record["FloatArray"]?.toFloatArray())
    print(record["Bytes"]?.toBytes())
    print(record["IntArray"]?.toIntArray())
}