2023-07-27 22:01:33 +02:00
|
|
|
{ // Convolution layer ---------------------------------------------------------
|
|
|
|
"conv_out_channel": [
|
|
|
|
32,
|
|
|
|
8,
|
|
|
|
8
|
|
|
|
],
|
|
|
|
"conv_kernel_size": [
|
|
|
|
11,
|
2023-07-28 00:11:46 +02:00
|
|
|
11,
|
|
|
|
11
|
2023-07-27 22:01:33 +02:00
|
|
|
],
|
|
|
|
"conv_stride_size": [
|
|
|
|
1,
|
|
|
|
1,
|
|
|
|
1
|
|
|
|
],
|
|
|
|
"conv_bias": [ // Warning: One more for the output layer
|
|
|
|
true,
|
|
|
|
true,
|
|
|
|
true,
|
|
|
|
true
|
|
|
|
],
|
|
|
|
"conv_padding": [
|
|
|
|
0,
|
|
|
|
0,
|
|
|
|
0
|
|
|
|
],
|
|
|
|
// Activation function -----------------------------------------------------
|
|
|
|
"activation_function": "leaky relu", // tanh, relu, (leaky relu), none
|
|
|
|
"l_relu_negative_slope": 0.1, // (0.1)
|
|
|
|
// Pooling layer -----------------------------------------------------------
|
|
|
|
"pooling_kernel_size": [
|
2023-07-28 00:11:46 +02:00
|
|
|
5,
|
|
|
|
5,
|
|
|
|
5
|
2023-07-27 22:01:33 +02:00
|
|
|
],
|
|
|
|
"pooling_stride": [
|
|
|
|
2,
|
2023-07-28 00:11:46 +02:00
|
|
|
2,
|
|
|
|
2
|
2023-07-27 22:01:33 +02:00
|
|
|
],
|
|
|
|
"pooling_type": "max", // (max), average, none
|
|
|
|
// Softmax layer -----------------------------------------------------------
|
|
|
|
"softmax_enable": [
|
|
|
|
false,
|
|
|
|
false,
|
|
|
|
false
|
|
|
|
],
|
|
|
|
"softmax_power": 0.0, // (0.0) = Exp
|
|
|
|
"softmax_meanmode": true, // true, false
|
|
|
|
"softmax_no_input_mode": false, // true, (false)
|
|
|
|
// Load pre-trained weights and biases -------------------------------------
|
|
|
|
"path_pretrained_weights_bias": "",
|
|
|
|
"train_weights": [ // Warning: One more for the output layer
|
|
|
|
true,
|
|
|
|
true,
|
|
|
|
true,
|
|
|
|
true
|
|
|
|
],
|
|
|
|
"train_bias": [ // Warning: One more for the output layer
|
|
|
|
true,
|
|
|
|
true,
|
|
|
|
true,
|
|
|
|
true
|
|
|
|
],
|
|
|
|
// Output layer ------------------------------------------------------------
|
|
|
|
"number_of_classes": 2
|
|
|
|
}
|