private void ConvertInputLayer(InputLayer layer, ConvertContext context)
{
context.Layers.Add(new ScriptInputLayerConfig
{
Name = layer.Name,
Dimensions = layer.Output.Dimensions.ToNHWC().ToArray(),
Output = context.AddOutput(layer.Output)
});
}
private void ConvertK210SeparableConv2d(K210SeparableConv2d layer, ConvertContext context)
{
context.Layers.Add(new ScriptSeparableConv2dLayerConfig
{
KernelSize = layer.KernelWidth,
Stride = GetStride(layer.PoolType),
Filters = layer.OutputChannels,
Activation = layer.FusedActivationFunction,
Input = context.Outputs[layer.Input.Connection.From],
Output = context.AddOutput(layer.Output)
});
}
private void ConvertK210Conv2d(K210Conv2d layer, ConvertContext context)
{
if (layer.Conv2dType != K210Conv2dType.Conv2d)
{
throw new NotSupportedException("Depthwise conv2d is not supported.");
}
context.Layers.Add(new ScriptConv2dLayerConfig
{
KernelSize = layer.KernelWidth,
Stride = GetStride(layer.PoolType),
Filters = layer.OutputChannels,
Activation = layer.FusedActivationFunction,
Input = context.Outputs[layer.Input.Connection.From],
Output = context.AddOutput(layer.Output)
});
}
