小编Lor*_*ano的帖子

如何在CoreML自定义层中实现ktf.image.resize_images?

我正在尝试在CoreML中实现一个自定义层,它解决了ktf.image.resize_images在Swift中作为自定义图层类执行功能的Lambda层.

这是我的Phyton脚本:

def resizeImage(x, size):
    return ktf.image.resize_images(x, size)

def convert_lambda(layer):
    if layer.function == resizeImage:
        params = NeuralNetwork_pb2.CustomLayerParams()

        params.className = "resizeImage"
        params.description = "Decoder Resizing"

        params.parameters["scale"].intValue = layer.arguments["size"][0].value

        print("LAMBDA CONVERSION = Size embedded to CoreML Model: %d" % layer.arguments["size"][0].value)

        return params
    else:
        return None

...

for i in range(decoder_n):
    strides = 1
    reverse_i = decoder_n - i - 1
    size = encoder_layers[decoder_n - i - 1].shape[1:3]
    out_channels = 2 ** ((decoder_n - i - 2) // 3 + 5) …
Run Code Online (Sandbox Code Playgroud)

swift coreml coremltools

9
推荐指数
0
解决办法
449
查看次数

张量流中的“有状态对象”是什么?

在文档的某些部分(例如,Dataset Iterators 此处)中有对的引用Stateful Objects。它们究竟是什么?它们在图中起什么作用?

要澄清的是,在数据集文档中,有一个有效的示例,one_shot_iterator因为它是无状态的:

dataset = tf.data.Dataset.range(100)
iterator = dataset.make_one_shot_iterator()
Run Code Online (Sandbox Code Playgroud)

是什么使迭代器无状态?

python tensorflow

3
推荐指数
1
解决办法
701
查看次数

标签 统计

coreml ×1

coremltools ×1

python ×1

swift ×1

tensorflow ×1