List <CNTK.Variable> traverse_content_and_styles_nodes(CNTK.Function model)
{
var nodes = new List <CNTK.Variable>();
var node_names = new string[] { "conv5_2", "conv1_1", "conv2_1", "conv3_1", "conv4_1", "conv5_1" };
foreach (var node_name in node_names)
{
var node = model.FindByName(node_name);
nodes.Add(node);
}
return(nodes);
}
/// <summary>
/// Get a list of content and style layers from the given neural network.
/// </summary>
/// <param name="input">The neural network to process.</param>
/// <returns>A list of content and style layers in the neural network.</returns>
public static List <CNTK.Variable> GetContentAndStyleLayers(
this CNTK.Function input)
{
var layers = new List <CNTK.Variable>();
var layerNames = new string[] { "conv5_2", "conv1_1", "conv2_1", "conv3_1", "conv4_1", "conv5_1" };
foreach (var layerName in layerNames)
{
var layer = input.FindByName(layerName);
layers.Add(layer);
}
return(layers);
}
