|
- using System;
- using System.Collections.Generic;
- using System.Text;
- using Tensorflow;
- using static Tensorflow.Binding;
-
- namespace TensorFlowNET.Examples.ImageProcessing.YOLO
- {
- class backbone
- {
- public static (Tensor, Tensor, Tensor) darknet53(Tensor input_data, Tensor trainable)
- {
- return tf_with(tf.variable_scope("darknet"), scope =>
- {
- input_data = common.convolutional(input_data, filters_shape: new int[] { 3, 3, 3, 32 }, trainable: trainable, name: "conv0");
- input_data = common.convolutional(input_data, filters_shape: new int[] { 3, 3, 32, 64 }, trainable: trainable, name: "conv1", downsample: true);
-
- foreach (var i in range(1))
- input_data = common.residual_block(input_data, 64, 32, 64, trainable: trainable, name: $"residual{i + 0}");
-
- var route_1 = input_data;
- var route_2 = input_data;
-
- return (route_1, route_2, input_data);
- });
- }
- }
- }
|