Gitiles
Code Review Sign In
gerrit-public.fairphone.software / platform / external / ComputeLibrary / 7f09cf7dfba61aad6d6d588b1b096f56bafaad5e / . / examples / graph_inception_v3.cpp
blob: 1de6a5fad75b11ab67bc5812750bae7db0c4d91a [file] [log] [blame]
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]1/*
Jenkins4ba87db2019-05-23 17:11:51 +0100[diff] [blame]2 * Copyright (c) 2017-2019 ARM Limited.
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]3 *
4 * SPDX-License-Identifier: MIT
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to
8 * deal in the Software without restriction, including without limitation the
9 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10 * sell copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in all
14 * copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22 * SOFTWARE.
23 */
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]24#include "arm_compute/graph.h"
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]25#include "support/ToolchainSupport.h"
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]26#include "utils/CommonGraphOptions.h"
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]27#include "utils/GraphUtils.h"
28#include "utils/Utils.h"
29
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]30using namespace arm_compute::utils;
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]31using namespace arm_compute::graph::frontend;
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]32using namespace arm_compute::graph_utils;
33
Jenkinsb9abeae2018-11-22 11:58:08 +0000[diff] [blame]34/** Example demonstrating how to implement InceptionV3's network using the Compute Library's graph API */
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]35class InceptionV3Example : public Example
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]36{
37public:
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]38 InceptionV3Example()
39 : cmd_parser(), common_opts(cmd_parser), common_params(), graph(0, "InceptionV3")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]40 {
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]41 }
42 bool do_setup(int argc, char **argv) override
43 {
44 // Parse arguments
45 cmd_parser.parse(argc, argv);
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]46 cmd_parser.validate();
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]47
48 // Consume common parameters
49 common_params = consume_common_graph_parameters(common_opts);
50
51 // Return when help menu is requested
52 if(common_params.help)
53 {
54 cmd_parser.print_help(argv[0]);
55 return false;
56 }
57
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]58 // Checks
59 ARM_COMPUTE_EXIT_ON_MSG(arm_compute::is_data_type_quantized_asymmetric(common_params.data_type), "QASYMM8 not supported for this graph");
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]60
61 // Print parameter values
62 std::cout << common_params << std::endl;
63
64 // Get trainable parameters data path
65 std::string data_path = common_params.data_path;
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]66
Anthony Barbier06ea0482018-02-22 15:45:35 +0000[diff] [blame]67 // Create a preprocessor object
68 std::unique_ptr<IPreprocessor> preprocessor = arm_compute::support::cpp14::make_unique<TFPreproccessor>();
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]69
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]70 // Create input descriptor
71 const TensorShape tensor_shape = permute_shape(TensorShape(299U, 299U, 3U, 1U), DataLayout::NCHW, common_params.data_layout);
72 TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_layout(common_params.data_layout);
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]73
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]74 // Set weights trained layout
75 const DataLayout weights_layout = DataLayout::NCHW;
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]76
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]77 graph << common_params.target
78 << common_params.fast_math_hint
79 << InputLayer(input_descriptor, get_input_accessor(common_params, std::move(preprocessor), false))
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]80 << ConvolutionLayer(3U, 3U, 32U,
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]81 get_weights_accessor(data_path, "/cnn_data/inceptionv3_model/Conv2d_1a_3x3_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]82 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr), PadStrideInfo(2, 2, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]83 .set_name("Conv2d_1a_3x3/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]84 << BatchNormalizationLayer(get_weights_accessor(data_path,
85 "/cnn_data/inceptionv3_model/Conv2d_1a_3x3_BatchNorm_moving_mean.npy"),
86 get_weights_accessor(data_path,
87 "/cnn_data/inceptionv3_model/Conv2d_1a_3x3_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]88 nullptr, get_weights_accessor(data_path,
89 "/cnn_data/inceptionv3_model/Conv2d_1a_3x3_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]90 0.001f)
91 .set_name("Conv2d_1a_3x3/BatchNorm/batchnorm")
92 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name("Conv2d_1a_3x3/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]93 << ConvolutionLayer(3U, 3U, 32U,
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]94 get_weights_accessor(data_path, "/cnn_data/inceptionv3_model/Conv2d_2a_3x3_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]95 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr), PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]96 .set_name("Conv2d_2a_3x3/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]97 << BatchNormalizationLayer(get_weights_accessor(data_path,
98 "/cnn_data/inceptionv3_model/Conv2d_2a_3x3_BatchNorm_moving_mean.npy"),
99 get_weights_accessor(data_path,
100 "/cnn_data/inceptionv3_model/Conv2d_2a_3x3_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]101 nullptr, get_weights_accessor(data_path,
102 "/cnn_data/inceptionv3_model/Conv2d_2a_3x3_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]103 0.001f)
104 .set_name("Conv2d_2a_3x3/BatchNorm/batchnorm")
105 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name("Conv2d_2a_3x3/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]106
107 << ConvolutionLayer(3U, 3U, 64U,
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]108 get_weights_accessor(data_path, "/cnn_data/inceptionv3_model/Conv2d_2b_3x3_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]109 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr), PadStrideInfo(1, 1, 1, 1))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]110 .set_name("Conv2d_2b_3x3/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]111 << BatchNormalizationLayer(get_weights_accessor(data_path,
112 "/cnn_data/inceptionv3_model/Conv2d_2b_3x3_BatchNorm_moving_mean.npy"),
113 get_weights_accessor(data_path,
114 "/cnn_data/inceptionv3_model/Conv2d_2b_3x3_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]115 nullptr, get_weights_accessor(data_path,
116 "/cnn_data/inceptionv3_model/Conv2d_2b_3x3_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]117 0.001f)
118 .set_name("Conv2d_2b_3x3/BatchNorm/batchnorm")
119 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name("Conv2d_2b_3x3/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]120
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]121 << PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2, 0, 0, DimensionRoundingType::CEIL))).set_name("MaxPool_3a_3x3/MaxPool")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]122
123 << ConvolutionLayer(1U, 1U, 80U,
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]124 get_weights_accessor(data_path, "/cnn_data/inceptionv3_model/Conv2d_3b_1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]125 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr), PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]126 .set_name("Conv2d_3b_1x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]127 << BatchNormalizationLayer(get_weights_accessor(data_path,
128 "/cnn_data/inceptionv3_model/Conv2d_3b_1x1_BatchNorm_moving_mean.npy"),
129 get_weights_accessor(data_path,
130 "/cnn_data/inceptionv3_model/Conv2d_3b_1x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]131 nullptr, get_weights_accessor(data_path,
132 "/cnn_data/inceptionv3_model/Conv2d_3b_1x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]133 0.001f)
134 .set_name("Conv2d_3b_1x1/BatchNorm/batchnorm")
135 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name("Conv2d_3b_1x1/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]136
137 << ConvolutionLayer(3U, 3U, 192U,
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]138 get_weights_accessor(data_path, "/cnn_data/inceptionv3_model/Conv2d_4a_3x3_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]139 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr), PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]140 .set_name("Conv2d_4a_3x3/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]141 << BatchNormalizationLayer(get_weights_accessor(data_path,
142 "/cnn_data/inceptionv3_model/Conv2d_4a_3x3_BatchNorm_moving_mean.npy"),
143 get_weights_accessor(data_path,
144 "/cnn_data/inceptionv3_model/Conv2d_4a_3x3_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]145 nullptr, get_weights_accessor(data_path,
146 "/cnn_data/inceptionv3_model/Conv2d_4a_3x3_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]147 0.001f)
148 .set_name("Conv2d_4a_3x3/BatchNorm/batchnorm")
149 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name("Conv2d_4a_3x3/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]150
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]151 << PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2, 0, 0, DimensionRoundingType::CEIL))).set_name("MaxPool_5a_3x3/MaxPool");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]152
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]153 graph << get_inception_node_A(data_path, "Mixed_5b", weights_layout, 64U, std::make_tuple(48U, 64U), std::make_tuple(64U, 96U, 96U),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]154 32U)
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]155 .set_name("Mixed_5b/concat");
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]156 graph << get_inception_node_A(data_path, "Mixed_5c", weights_layout, 64U, std::make_tuple(48U, 64U), std::make_tuple(64U, 96U, 96U),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]157 64U, true)
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]158 .set_name("Mixed_5c/concat");
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]159 graph << get_inception_node_A(data_path, "Mixed_5d", weights_layout, 64U, std::make_tuple(48U, 64U), std::make_tuple(64U, 96U, 96U),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]160 64U)
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]161 .set_name("Mixed_5d/concat");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]162
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]163 graph << get_inception_node_B(data_path, "Mixed_6a", weights_layout, 384U, std::make_tuple(64U, 96U, 96U)).set_name("Mixed_6a/concat");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]164
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]165 graph << get_inception_node_C(data_path, "Mixed_6b", weights_layout, 192U, std::make_tuple(128U, 128U, 192U),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]166 std::make_tuple(128U, 128U, 128U, 128U, 192U), 192U)
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]167 .set_name("Mixed_6b/concat");
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]168 graph << get_inception_node_C(data_path, "Mixed_6c", weights_layout, 192U, std::make_tuple(160U, 160U, 192U),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]169 std::make_tuple(160U, 160U, 160U, 160U, 192U), 192U)
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]170 .set_name("Mixed_6c/concat");
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]171 graph << get_inception_node_C(data_path, "Mixed_6d", weights_layout, 192U, std::make_tuple(160U, 160U, 192U),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]172 std::make_tuple(160U, 160U, 160U, 160U, 192U), 192U)
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]173 .set_name("Mixed_6d/concat");
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]174 graph << get_inception_node_C(data_path, "Mixed_6e", weights_layout, 192U, std::make_tuple(192U, 192U, 192U),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]175 std::make_tuple(192U, 192U, 192U, 192U, 192U), 192U)
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]176 .set_name("Mixed_6e/concat");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]177
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]178 graph << get_inception_node_D(data_path, "Mixed_7a", weights_layout, std::make_tuple(192U, 320U),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]179 std::make_tuple(192U, 192U, 192U, 192U))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]180 .set_name("Mixed_7a/concat");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]181
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]182 graph << get_inception_node_E(data_path, "Mixed_7b", weights_layout, 320U, std::make_tuple(384U, 384U, 384U),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]183 std::make_tuple(448U, 384U, 384U, 384U), 192U)
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]184 .set_name("Mixed_7b/concat");
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]185 graph << get_inception_node_E(data_path, "Mixed_7c", weights_layout, 320U, std::make_tuple(384U, 384U, 384U),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]186 std::make_tuple(448U, 384U, 384U, 384U), 192U, true)
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]187 .set_name("Mixed_7c/concat");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]188
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]189 graph << PoolingLayer(PoolingLayerInfo(PoolingType::AVG, 8, PadStrideInfo(1, 1, 0, 0, DimensionRoundingType::CEIL))).set_name("Logits/AvgPool_1a_8x8/AvgPool")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]190 << ConvolutionLayer(1U, 1U, 1001U, get_weights_accessor(data_path,
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]191 "/cnn_data/inceptionv3_model/Logits_Conv2d_1c_1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]192 get_weights_accessor(data_path,
193 "/cnn_data/inceptionv3_model/Logits_Conv2d_1c_1x1_biases.npy"),
194 PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]195 .set_name("Logits/Conv2d_1c_1x1/convolution")
196 << ReshapeLayer(TensorShape(1001U)).set_name("Predictions/Reshape")
197 << SoftmaxLayer().set_name("Predictions/Softmax")
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]198 << OutputLayer(get_output_accessor(common_params, 5));
Anthony Barbier06ea0482018-02-22 15:45:35 +0000[diff] [blame]199
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]200 // Finalize graph
201 GraphConfig config;
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]202 config.num_threads = common_params.threads;
203 config.use_tuner = common_params.enable_tuner;
Jenkins4ba87db2019-05-23 17:11:51 +0100[diff] [blame]204 config.tuner_mode = common_params.tuner_mode;
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]205 config.tuner_file = common_params.tuner_file;
206
207 graph.finalize(common_params.target, config);
208
209 return true;
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]210 }
211
212 void do_run() override
213 {
214 graph.run();
215 }
216
217private:
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]218 CommandLineParser cmd_parser;
219 CommonGraphOptions common_opts;
220 CommonGraphParams common_params;
221 Stream graph;
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]222
223private:
Jenkinsb9abeae2018-11-22 11:58:08 +0000[diff] [blame]224 ConcatLayer get_inception_node_A(const std::string &data_path, std::string &&param_path, DataLayout weights_layout,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]225 unsigned int a_filt,
226 std::tuple<unsigned int, unsigned int> b_filters,
227 std::tuple<unsigned int, unsigned int, unsigned int> c_filters,
228 unsigned int d_filt,
229 bool is_name_different = false)
230 {
231 std::string total_path = "/cnn_data/inceptionv3_model/" + param_path + "_";
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]232
233 // This is due to a naming issue in the tf model
234 std::string conv_id0 = "_0a_";
235 std::string conv_id1 = "2d_0b_";
236 if(is_name_different)
237 {
238 conv_id0 = "_0b_";
239 conv_id1 = "_1_0c_";
240 }
241
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]242 SubStream i_a(graph);
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]243 i_a << ConvolutionLayer(
244 1U, 1U, a_filt,
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]245 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_0a_1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]246 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
247 PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]248 .set_name(param_path + "/Branch_0/Conv2d_0a_1x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]249 << BatchNormalizationLayer(
250 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_0a_1x1_BatchNorm_moving_mean.npy"),
251 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_0a_1x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]252 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]253 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_0a_1x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]254 0.001f)
255 .set_name(param_path + "/Branch_0/Conv2d_0a_1x1/BatchNorm/batchnorm")
256 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_0/Conv2d_0a_1x1/Relu");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]257
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]258 SubStream i_b(graph);
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]259 i_b << ConvolutionLayer(
260 1U, 1U, std::get<0>(b_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]261 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d" + conv_id0 + "1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]262 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
263 PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]264 .set_name(param_path + "/Branch_1/Conv2d" + conv_id0 + "1x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]265 << BatchNormalizationLayer(
266 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d" + conv_id0 + "1x1_BatchNorm_moving_mean.npy"),
267 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d" + conv_id0 + "1x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]268 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]269 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d" + conv_id0 + "1x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]270 0.001f)
271 .set_name(param_path + "/Branch_1/Conv2d" + conv_id0 + "1x1/BatchNorm/batchnorm")
272 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_1/Conv2d" + conv_id0 + "1x1/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]273 << ConvolutionLayer(
274 5U, 5U, std::get<1>(b_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]275 get_weights_accessor(data_path, total_path + "Branch_1_Conv" + conv_id1 + "5x5_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]276 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
277 PadStrideInfo(1, 1, 2, 2))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]278 .set_name(param_path + "/Branch_1/Conv2d" + conv_id1 + "5x5/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]279 << BatchNormalizationLayer(
280 get_weights_accessor(data_path, total_path + "Branch_1_Conv" + conv_id1 + "5x5_BatchNorm_moving_mean.npy"),
281 get_weights_accessor(data_path, total_path + "Branch_1_Conv" + conv_id1 + "5x5_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]282 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]283 get_weights_accessor(data_path, total_path + "Branch_1_Conv" + conv_id1 + "5x5_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]284 0.001f)
285 .set_name(param_path + "/Branch_1/Conv2d" + conv_id1 + "5x5/BatchNorm/batchnorm")
286 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_1/Conv2d" + conv_id1 + "5x5/Relu");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]287
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]288 SubStream i_c(graph);
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]289 i_c << ConvolutionLayer(
290 1U, 1U, std::get<0>(c_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]291 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0a_1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]292 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
293 PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]294 .set_name(param_path + "/Branch_2/Conv2d_0a_1x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]295 << BatchNormalizationLayer(
296 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0a_1x1_BatchNorm_moving_mean.npy"),
297 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0a_1x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]298 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]299 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0a_1x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]300 0.001f)
301 .set_name(param_path + "/Branch_2/Conv2d_0a_1x1/BatchNorm/batchnorm")
302 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_2/Conv2d_0a_1x1/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]303 << ConvolutionLayer(
304 3U, 3U, std::get<1>(c_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]305 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0b_3x3_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]306 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
307 PadStrideInfo(1, 1, 1, 1))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]308 .set_name(param_path + "/Branch_2/Conv2d_0b_3x3/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]309 << BatchNormalizationLayer(
310 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0b_3x3_BatchNorm_moving_mean.npy"),
311 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0b_3x3_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]312 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]313 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0b_3x3_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]314 0.001f)
315 .set_name(param_path + "/Branch_2/Conv2d_0b_3x3/BatchNorm/batchnorm")
316 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_2/Conv2d_0b_3x3/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]317 << ConvolutionLayer(
318 3U, 3U, std::get<2>(c_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]319 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0c_3x3_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]320 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
321 PadStrideInfo(1, 1, 1, 1))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]322 .set_name(param_path + "/Branch_2/Conv2d_0c_3x3/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]323 << BatchNormalizationLayer(
324 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0c_3x3_BatchNorm_moving_mean.npy"),
325 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0c_3x3_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]326 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]327 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0c_3x3_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]328 0.001f)
329 .set_name(param_path + "/Branch_2/Conv2d_0c_3x3/BatchNorm/batcnorm")
330 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_2/Conv2d_0c_3x3/Relu");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]331
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]332 SubStream i_d(graph);
333 i_d << PoolingLayer(PoolingLayerInfo(PoolingType::AVG, 3, PadStrideInfo(1, 1, 1, 1, DimensionRoundingType::CEIL), true)).set_name(param_path + "/Branch_3/AvgPool_0a_3x3/AvgPool")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]334 << ConvolutionLayer(
335 1U, 1U, d_filt,
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]336 get_weights_accessor(data_path, total_path + "Branch_3_Conv2d_0b_1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]337 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
338 PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]339 .set_name(param_path + "/Branch_3/Conv2d_0b_1x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]340 << BatchNormalizationLayer(
341 get_weights_accessor(data_path, total_path + "Branch_3_Conv2d_0b_1x1_BatchNorm_moving_mean.npy"),
342 get_weights_accessor(data_path, total_path + "Branch_3_Conv2d_0b_1x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]343 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]344 get_weights_accessor(data_path, total_path + "Branch_3_Conv2d_0b_1x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]345 0.001f)
346 .set_name(param_path + "/Branch_3/Conv2d_0b_1x1/BatchNorm/batchnorm")
347 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_3/Conv2d_0b_1x1/Relu");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]348
Jenkinsb9abeae2018-11-22 11:58:08 +0000[diff] [blame]349 return ConcatLayer(std::move(i_a), std::move(i_b), std::move(i_c), std::move(i_d));
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]350 }
351
Jenkinsb9abeae2018-11-22 11:58:08 +0000[diff] [blame]352 ConcatLayer get_inception_node_B(const std::string &data_path, std::string &&param_path, DataLayout weights_layout,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]353 unsigned int a_filt,
354 std::tuple<unsigned int, unsigned int, unsigned int> b_filters)
355 {
356 std::string total_path = "/cnn_data/inceptionv3_model/" + param_path + "_";
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]357 SubStream i_a(graph);
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]358 i_a << ConvolutionLayer(
359 3U, 3U, a_filt,
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]360 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_1a_1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]361 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
362 PadStrideInfo(2, 2, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]363 .set_name(param_path + "/Branch_0/Conv2d_1a_1x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]364 << BatchNormalizationLayer(
365 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_1a_1x1_BatchNorm_moving_mean.npy"),
366 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_1a_1x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]367 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]368 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_1a_1x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]369 0.001f)
370 .set_name(param_path + "/Branch_0/Conv2d_1a_1x1/BatchNorm/batchnorm")
371 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_0/Conv2d_1a_1x1/Relu");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]372
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]373 SubStream i_b(graph);
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]374 i_b << ConvolutionLayer(
375 1U, 1U, std::get<0>(b_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]376 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0a_1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]377 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
378 PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]379 .set_name(param_path + "/Branch_1/Conv2d_0a_1x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]380 << BatchNormalizationLayer(
381 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0a_1x1_BatchNorm_moving_mean.npy"),
382 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0a_1x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]383 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]384 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0a_1x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]385 0.001f)
386 .set_name(param_path + "/Branch_1/Conv2d_0a_1x1/BatchNorm/batchnorm")
387 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_1/Conv2d_0a_1x1/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]388 << ConvolutionLayer(
389 3U, 3U, std::get<1>(b_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]390 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0b_3x3_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]391 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
392 PadStrideInfo(1, 1, 1, 1))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]393 .set_name(param_path + "/Branch_1/Conv2d_0b_3x3/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]394 << BatchNormalizationLayer(
395 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0b_3x3_BatchNorm_moving_mean.npy"),
396 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0b_3x3_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]397 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]398 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0b_3x3_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]399 0.001f)
400 .set_name(param_path + "/Branch_1/Conv2d_0b_3x3/BatchNorm/batchnorm")
401 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_1/Conv2d_0b_3x3/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]402 << ConvolutionLayer(
403 3U, 3U, std::get<2>(b_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]404 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_1a_1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]405 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
406 PadStrideInfo(2, 2, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]407 .set_name(param_path + "/Branch_1/Conv2d_1a_1x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]408 << BatchNormalizationLayer(
409 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_1a_1x1_BatchNorm_moving_mean.npy"),
410 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_1a_1x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]411 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]412 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_1a_1x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]413 0.001f)
414 .set_name(param_path + "/Branch_1/Conv2d_1a_1x1/BatchNorm/batchnorm")
415 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_1/Conv2d_1a_1x1/Relu");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]416
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]417 SubStream i_c(graph);
418 i_c << PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2, 0, 0, DimensionRoundingType::CEIL))).set_name(param_path + "/Branch_2/MaxPool_1a_3x3/MaxPool");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]419
Jenkinsb9abeae2018-11-22 11:58:08 +0000[diff] [blame]420 return ConcatLayer(std::move(i_a), std::move(i_b), std::move(i_c));
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]421 }
422
Jenkinsb9abeae2018-11-22 11:58:08 +0000[diff] [blame]423 ConcatLayer get_inception_node_C(const std::string &data_path, std::string &&param_path, DataLayout weights_layout,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]424 unsigned int a_filt,
425 std::tuple<unsigned int, unsigned int, unsigned int> b_filters,
426 std::tuple<unsigned int, unsigned int, unsigned int, unsigned int, unsigned int> c_filters,
427 unsigned int d_filt)
428 {
429 std::string total_path = "/cnn_data/inceptionv3_model/" + param_path + "_";
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]430 SubStream i_a(graph);
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]431 i_a << ConvolutionLayer(
432 1U, 1U, a_filt,
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]433 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_0a_1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]434 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
435 PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]436 .set_name(param_path + "/Branch_0/Conv2d_0a_1x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]437 << BatchNormalizationLayer(
438 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_0a_1x1_BatchNorm_moving_mean.npy"),
439 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_0a_1x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]440 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]441 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_0a_1x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]442 0.001f)
443 .set_name(param_path + "/Branch_0/Conv2d_0a_1x1/BatchNorm/batchnorm")
444 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_0/Conv2d_0a_1x1/Relu");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]445
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]446 SubStream i_b(graph);
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]447 i_b << ConvolutionLayer(
448 1U, 1U, std::get<0>(b_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]449 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0a_1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]450 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
451 PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]452 .set_name(param_path + "/Branch_1/Conv2d_0a_1x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]453 << BatchNormalizationLayer(
454 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0a_1x1_BatchNorm_moving_mean.npy"),
455 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0a_1x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]456 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]457 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0a_1x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]458 0.001f)
459 .set_name(param_path + "/Branch_1/Conv2d_0a_1x1/BatchNorm/batchnorm")
460 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_1/Conv2d_0a_1x1/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]461 << ConvolutionLayer(
462 7U, 1U, std::get<1>(b_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]463 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0b_1x7_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]464 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
465 PadStrideInfo(1, 1, 3, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]466 .set_name(param_path + "/Branch_1/Conv2d_0b_1x7/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]467 << BatchNormalizationLayer(
468 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0b_1x7_BatchNorm_moving_mean.npy"),
469 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0b_1x7_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]470 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]471 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0b_1x7_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]472 0.001f)
473 .set_name(param_path + "/Branch_1/Conv2d_0b_1x7/BatchNorm/batchnorm")
474 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_1/Conv2d_0b_1x7/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]475 << ConvolutionLayer(
476 1U, 7U, std::get<2>(b_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]477 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0c_7x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]478 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
479 PadStrideInfo(1, 1, 0, 3))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]480 .set_name(param_path + "/Branch_1/Conv2d_0c_7x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]481 << BatchNormalizationLayer(
482 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0c_7x1_BatchNorm_moving_mean.npy"),
483 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0c_7x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]484 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]485 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0c_7x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]486 0.001f)
487 .set_name(param_path + "/Branch_1/Conv2d_0c_7x1/BatchNorm/batchnorm")
488 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_0/Conv2d_0c_7x1/Relu");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]489
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]490 SubStream i_c(graph);
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]491 i_c << ConvolutionLayer(
492 1U, 1U, std::get<0>(c_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]493 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0a_1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]494 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
495 PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]496 .set_name(param_path + "/Branch_2/Conv2d_0a_1x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]497 << BatchNormalizationLayer(
498 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0a_1x1_BatchNorm_moving_mean.npy"),
499 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0a_1x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]500 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]501 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0a_1x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]502 0.001f)
503 .set_name(param_path + "/Branch_2/Conv2d_0a_1x1/BatchNorm/batchnorm")
504 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_2/Conv2d_0a_1x1/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]505 << ConvolutionLayer(
506 1U, 7U, std::get<1>(c_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]507 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0b_7x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]508 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
509 PadStrideInfo(1, 1, 0, 3))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]510 .set_name(param_path + "/Branch_2/Conv2d_0b_7x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]511 << BatchNormalizationLayer(
512 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0b_7x1_BatchNorm_moving_mean.npy"),
513 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0b_7x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]514 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]515 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0b_7x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]516 0.001f)
517 .set_name(param_path + "/Branch_2/Conv2d_0b_7x1/BatchNorm/batchnorm")
518 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_2/Conv2d_0b_7x1/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]519 << ConvolutionLayer(
520 7U, 1U, std::get<2>(c_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]521 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0c_1x7_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]522 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
523 PadStrideInfo(1, 1, 3, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]524 .set_name(param_path + "/Branch_2/Conv2d_0c_1x7/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]525 << BatchNormalizationLayer(
526 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0c_1x7_BatchNorm_moving_mean.npy"),
527 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0c_1x7_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]528 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]529 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0c_1x7_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]530 0.001f)
531 .set_name(param_path + "/Branch_2/Conv2d_0c_1x7/BatchNorm/batchnorm")
532 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_2/Conv2d_0c_1x7/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]533 << ConvolutionLayer(
534 1U, 7U, std::get<3>(c_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]535 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0d_7x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]536 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
537 PadStrideInfo(1, 1, 0, 3))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]538 .set_name(param_path + "/Branch_2/Conv2d_0d_7x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]539 << BatchNormalizationLayer(
540 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0d_7x1_BatchNorm_moving_mean.npy"),
541 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0d_7x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]542 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]543 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0d_7x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]544 0.001f)
545 .set_name(param_path + "/Branch_2/Conv2d_0d_7x1/BatchNorm/batchnorm")
546 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_2/Conv2d_0d_7x1/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]547 << ConvolutionLayer(
548 7U, 1U, std::get<4>(c_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]549 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0e_1x7_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]550 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
551 PadStrideInfo(1, 1, 3, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]552 .set_name(param_path + "/Branch_2/Conv2d_0e_1x7/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]553 << BatchNormalizationLayer(
554 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0e_1x7_BatchNorm_moving_mean.npy"),
555 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0e_1x7_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]556 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]557 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0e_1x7_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]558 0.001f)
559 .set_name(param_path + "/Branch_2/Conv2d_0e_1x7/BatchNorm/batchnorm")
560 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_2/Conv2d_0e_1x7/Relu");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]561
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]562 SubStream i_d(graph);
563 i_d << PoolingLayer(PoolingLayerInfo(PoolingType::AVG, 3, PadStrideInfo(1, 1, 1, 1, DimensionRoundingType::CEIL), true)).set_name(param_path + "/Branch_3/AvgPool_0a_3x3/AvgPool")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]564 << ConvolutionLayer(
565 1U, 1U, d_filt,
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]566 get_weights_accessor(data_path, total_path + "Branch_3_Conv2d_0b_1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]567 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
568 PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]569 .set_name(param_path + "/Branch_3/Conv2d_0b_1x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]570 << BatchNormalizationLayer(
571 get_weights_accessor(data_path, total_path + "Branch_3_Conv2d_0b_1x1_BatchNorm_moving_mean.npy"),
572 get_weights_accessor(data_path, total_path + "Branch_3_Conv2d_0b_1x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]573 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]574 get_weights_accessor(data_path, total_path + "Branch_3_Conv2d_0b_1x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]575 0.001f)
576 .set_name(param_path + "/Branch_3/Conv2d_0b_1x1/BatchNorm/batchnorm")
577 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_3/Conv2d_0b_1x1/Relu");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]578
Jenkinsb9abeae2018-11-22 11:58:08 +0000[diff] [blame]579 return ConcatLayer(std::move(i_a), std::move(i_b), std::move(i_c), std::move(i_d));
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]580 }
581
Jenkinsb9abeae2018-11-22 11:58:08 +0000[diff] [blame]582 ConcatLayer get_inception_node_D(const std::string &data_path, std::string &&param_path, DataLayout weights_layout,
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]583 std::tuple<unsigned int, unsigned int> a_filters,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]584 std::tuple<unsigned int, unsigned int, unsigned int, unsigned int> b_filters)
585 {
586 std::string total_path = "/cnn_data/inceptionv3_model/" + param_path + "_";
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]587 SubStream i_a(graph);
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]588 i_a << ConvolutionLayer(
589 1U, 1U, std::get<0>(a_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]590 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_0a_1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]591 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
592 PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]593 .set_name(param_path + "/Branch_0/Conv2d_0a_1x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]594 << BatchNormalizationLayer(
595 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_0a_1x1_BatchNorm_moving_mean.npy"),
596 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_0a_1x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]597 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]598 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_0a_1x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]599 0.001f)
600 .set_name(param_path + "/Branch_0/Conv2d_0a_1x1/BatchNorm/batchnorm")
601 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_0/Conv2d_0a_1x1/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]602 << ConvolutionLayer(
603 3U, 3U, std::get<1>(a_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]604 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_1a_3x3_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]605 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
606 PadStrideInfo(2, 2, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]607 .set_name(param_path + "/Branch_0/Conv2d_1a_3x3/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]608 << BatchNormalizationLayer(
609 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_1a_3x3_BatchNorm_moving_mean.npy"),
610 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_1a_3x3_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]611 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]612 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_1a_3x3_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]613 0.001f)
614 .set_name(param_path + "/Branch_0/Conv2d_1a_3x3/BatchNorm/batchnorm")
615 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_0/Conv2d_1a_3x3/Relu");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]616
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]617 SubStream i_b(graph);
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]618 i_b << ConvolutionLayer(
619 1U, 1U, std::get<0>(b_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]620 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0a_1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]621 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
622 PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]623 .set_name(param_path + "/Branch_1/Conv2d_0a_1x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]624 << BatchNormalizationLayer(
625 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0a_1x1_BatchNorm_moving_mean.npy"),
626 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0a_1x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]627 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]628 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0a_1x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]629 0.001f)
630 .set_name(param_path + "/Branch_1/Conv2d_0a_1x1/BatchNorm/batchnorm")
631 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_1/Conv2d_0a_1x1/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]632 << ConvolutionLayer(
633 7U, 1U, std::get<1>(b_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]634 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0b_1x7_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]635 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
636 PadStrideInfo(1, 1, 3, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]637 .set_name(param_path + "/Branch_1/Conv2d_0b_1x7/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]638 << BatchNormalizationLayer(
639 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0b_1x7_BatchNorm_moving_mean.npy"),
640 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0b_1x7_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]641 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]642 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0b_1x7_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]643 0.001f)
644 .set_name(param_path + "/Branch_1/Conv2d_0b_1x7/BatchNorm/batchnorm")
645 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_1/Conv2d_0b_1x7/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]646 << ConvolutionLayer(
647 1U, 7U, std::get<2>(b_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]648 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0c_7x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]649 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
650 PadStrideInfo(1, 1, 0, 3))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]651 .set_name(param_path + "/Branch_1/Conv2d_0c_7x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]652 << BatchNormalizationLayer(
653 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0c_7x1_BatchNorm_moving_mean.npy"),
654 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0c_7x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]655 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]656 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0c_7x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]657 0.001f)
658 .set_name(param_path + "/Branch_1/Conv2d_0c_7x1/BatchNorm/batchnorm")
659 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_1/Conv2d_0c_7x1/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]660 << ConvolutionLayer(
661 3U, 3U, std::get<3>(b_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]662 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_1a_3x3_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]663 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
664 PadStrideInfo(2, 2, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]665 .set_name(param_path + "/Branch_1/Conv2d_1a_3x3/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]666 << BatchNormalizationLayer(
667 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_1a_3x3_BatchNorm_moving_mean.npy"),
668 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_1a_3x3_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]669 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]670 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_1a_3x3_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]671 0.001f)
672 .set_name(param_path + "/Branch_1/Conv2d_1a_3x3/BatchNorm/batchnorm")
673 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_1/Conv2d_1a_3x3/Relu");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]674
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]675 SubStream i_c(graph);
676 i_c << PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2, 0, 0, DimensionRoundingType::CEIL))).set_name(param_path + "/Branch_2/MaxPool_1a_3x3/MaxPool");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]677
Jenkinsb9abeae2018-11-22 11:58:08 +0000[diff] [blame]678 return ConcatLayer(std::move(i_a), std::move(i_b), std::move(i_c));
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]679 }
680
Jenkinsb9abeae2018-11-22 11:58:08 +0000[diff] [blame]681 ConcatLayer get_inception_node_E(const std::string &data_path, std::string &&param_path, DataLayout weights_layout,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]682 unsigned int a_filt,
683 std::tuple<unsigned int, unsigned int, unsigned int> b_filters,
684 std::tuple<unsigned int, unsigned int, unsigned int, unsigned int> c_filters,
685 unsigned int d_filt,
686 bool is_name_different = false)
687 {
688 // This is due to a naming issue in the tf model
689 std::string conv_id = "_0b_";
690 if(is_name_different)
691 {
692 conv_id = "_0c_";
693 }
694
695 std::string total_path = "/cnn_data/inceptionv3_model/" + param_path + "_";
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]696 SubStream i_a(graph);
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]697 i_a << ConvolutionLayer(
698 1U, 1U, a_filt,
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]699 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_0a_1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]700 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
701 PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]702 .set_name(param_path + "/Branch_0/Conv2d_0a_1x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]703 << BatchNormalizationLayer(
704 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_0a_1x1_BatchNorm_moving_mean.npy"),
705 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_0a_1x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]706 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]707 get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_0a_1x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]708 0.001f)
709 .set_name(param_path + "/Branch_0/Conv2d_0a_1x1/BatchNorm/batchnorm")
710 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_0/Conv2d_0a_1x1/Relu");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]711
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]712 SubStream i_b(graph);
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]713 i_b << ConvolutionLayer(
714 1U, 1U, std::get<0>(b_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]715 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0a_1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]716 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
717 PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]718 .set_name(param_path + "/Branch_1/Conv2d_0a_1x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]719 << BatchNormalizationLayer(
720 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0a_1x1_BatchNorm_moving_mean.npy"),
721 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0a_1x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]722 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]723 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0a_1x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]724 0.001f)
725 .set_name(param_path + "/Branch_1/Conv2d_0a_1x1/BatchNorm/batchnorm")
726 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_1/Conv2d_0a_1x1/Relu");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]727
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]728 SubStream i_b1(i_b);
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]729 i_b1 << ConvolutionLayer(
730 3U, 1U, std::get<1>(b_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]731 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0b_1x3_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]732 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
733 PadStrideInfo(1, 1, 1, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]734 .set_name(param_path + "/Branch_1/Conv2d_0b_1x3/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]735 << BatchNormalizationLayer(
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]736 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0b_1x3_BatchNorm_moving_mean.npy"),
737 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0b_1x3_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]738 nullptr,
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]739 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0b_1x3_BatchNorm_beta.npy"),
740 0.001f)
741 .set_name(param_path + "/Branch_1/Conv2d_0b_1x3/BatchNorm/batchnorm")
742 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_1/Conv2d_0b_1x3/Relu");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]743
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]744 SubStream i_b2(i_b);
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]745 i_b2 << ConvolutionLayer(
746 1U, 3U, std::get<2>(b_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]747 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d" + conv_id + "3x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]748 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
749 PadStrideInfo(1, 1, 0, 1))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]750 .set_name(param_path + "/Branch_1/Conv2d" + conv_id + "3x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]751 << BatchNormalizationLayer(
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]752 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d" + conv_id + "3x1_BatchNorm_moving_mean.npy"),
753 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d" + conv_id + "3x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]754 nullptr,
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]755 get_weights_accessor(data_path, total_path + "Branch_1_Conv2d" + conv_id + "3x1_BatchNorm_beta.npy"),
756 0.001f)
757 .set_name(param_path + "/Branch_1/Conv2d" + conv_id + "3x1/BatchNorm/batchnorm")
758 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_1/Conv2d" + conv_id + "3x1/Relu");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]759
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]760 // Merge b1 and b2
Jenkinsb9abeae2018-11-22 11:58:08 +0000[diff] [blame]761 i_b << ConcatLayer(std::move(i_b1), std::move(i_b2)).set_name(param_path + "/Branch_1/concat");
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]762
763 SubStream i_c(graph);
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]764 i_c << ConvolutionLayer(
765 1U, 1U, std::get<0>(c_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]766 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0a_1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]767 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
768 PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]769 .set_name(param_path + "/Branch_2/Conv2d_0a_1x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]770 << BatchNormalizationLayer(
771 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0a_1x1_BatchNorm_moving_mean.npy"),
772 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0a_1x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]773 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]774 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0a_1x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]775 0.001f)
776 .set_name(param_path + "/Branch_2/Conv2d_0a_1x1/BatchNorm/batchnorm")
777 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_2/Conv2d_0a_1x1/Relu")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]778 << ConvolutionLayer(
779 3U, 3U, std::get<1>(c_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]780 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0b_3x3_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]781 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
782 PadStrideInfo(1, 1, 1, 1))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]783 .set_name(param_path + "/Branch_2/Conv2d_0b_3x3/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]784 << BatchNormalizationLayer(
785 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0b_3x3_BatchNorm_moving_mean.npy"),
786 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0b_3x3_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]787 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]788 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0b_3x3_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]789 0.001f)
790 .set_name(param_path + "/Branch_2/Conv2d_0b_3x3/BatchNorm/batchnorm")
791 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_2/Conv2d_0b_3x3/Relu");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]792
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]793 SubStream i_c1(i_c);
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]794 i_c1 << ConvolutionLayer(
795 3U, 1U, std::get<2>(c_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]796 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0c_1x3_weights.npy", weights_layout),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]797 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
798 PadStrideInfo(1, 1, 1, 0))
799 .set_name(param_path + "/Branch_2/Conv2d_0c_1x3/convolution")
800 << BatchNormalizationLayer(
801 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0c_1x3_BatchNorm_moving_mean.npy"),
802 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0c_1x3_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]803 nullptr,
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]804 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0c_1x3_BatchNorm_beta.npy"),
805 0.001f)
806 .set_name(param_path + "/Branch_2/Conv2d_0c_1x3/BatchNorm/batchnorm")
807 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_2/Conv2d_0c_1x3/Relu");
808
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]809 SubStream i_c2(i_c);
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]810 i_c2 << ConvolutionLayer(
811 1U, 3U, std::get<3>(c_filters),
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]812 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0d_3x1_weights.npy", weights_layout),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]813 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
814 PadStrideInfo(1, 1, 0, 1))
815 .set_name(param_path + "/Branch_2/Conv2d_0d_3x1/convolution")
816 << BatchNormalizationLayer(
817 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0d_3x1_BatchNorm_moving_mean.npy"),
818 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0d_3x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]819 nullptr,
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]820 get_weights_accessor(data_path, total_path + "Branch_2_Conv2d_0d_3x1_BatchNorm_beta.npy"),
821 0.001f)
822 .set_name(param_path + "/Branch_2/Conv2d_0d_3x1/BatchNorm/batchnorm")
823 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_2/Conv2d_0d_3x1/Relu");
824
825 // Merge i_c1 and i_c2
Jenkinsb9abeae2018-11-22 11:58:08 +0000[diff] [blame]826 i_c << ConcatLayer(std::move(i_c1), std::move(i_c2)).set_name(param_path + "/Branch_2/concat");
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]827
828 SubStream i_d(graph);
829 i_d << PoolingLayer(PoolingLayerInfo(PoolingType::AVG, 3, PadStrideInfo(1, 1, 1, 1, DimensionRoundingType::CEIL), true)).set_name(param_path + "/Branch_3/AvgPool_0a_3x3/AvgPool")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]830 << ConvolutionLayer(
831 1U, 1U, d_filt,
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]832 get_weights_accessor(data_path, total_path + "Branch_3_Conv2d_0b_1x1_weights.npy", weights_layout),
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]833 std::unique_ptr<arm_compute::graph::ITensorAccessor>(nullptr),
834 PadStrideInfo(1, 1, 0, 0))
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]835 .set_name(param_path + "/Branch_3/Conv2d_0b_1x1/convolution")
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]836 << BatchNormalizationLayer(
837 get_weights_accessor(data_path, total_path + "Branch_3_Conv2d_0b_1x1_BatchNorm_moving_mean.npy"),
838 get_weights_accessor(data_path, total_path + "Branch_3_Conv2d_0b_1x1_BatchNorm_moving_variance.npy"),
Jenkins0e205f72019-11-28 16:53:35 +0000[diff] [blame]839 nullptr,
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]840 get_weights_accessor(data_path, total_path + "Branch_3_Conv2d_0b_1x1_BatchNorm_beta.npy"),
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]841 0.001f)
842 .set_name(param_path + "/Branch_3/Conv2d_0b_1x1/BatchNorm/batchnorm")
843 << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(param_path + "/Branch_3/Conv2d_0b_1x1/Relu");
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]844
Jenkinsb9abeae2018-11-22 11:58:08 +0000[diff] [blame]845 return ConcatLayer(std::move(i_a), std::move(i_b), std::move(i_c), std::move(i_d));
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]846 }
847};
848
849/** Main program for Inception V3
850 *
Jenkinsb9abeae2018-11-22 11:58:08 +0000[diff] [blame]851 * Model is based on:
852 * https://arxiv.org/abs/1512.00567
853 * "Rethinking the Inception Architecture for Computer Vision"
854 * Christian Szegedy, Vincent Vanhoucke, Sergey Ioffe, Jonathon Shlens, Zbigniew Wojna
855 *
Jenkins514be652019-02-28 12:25:18 +0000[diff] [blame]856 * Provenance: download.tensorflow.org/models/inception_v3_2016_08_28.tar.gz
857 *
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]858 * @note To list all the possible arguments execute the binary appended with the --help option
859 *
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]860 * @param[in] argc Number of arguments
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame]861 * @param[in] argv Arguments
Anthony Barbierf45d5a92018-01-24 16:23:15 +0000[diff] [blame]862 */
863int main(int argc, char **argv)
864{
865 return arm_compute::utils::run_example<InceptionV3Example>(argc, argv);
866}