36 class GraphVDSRExample :
public Example 40 : cmd_parser(), common_opts(cmd_parser), common_params(), graph(0,
"VDSR")
46 model_input_width->
set_help(
"Input image width.");
47 model_input_height->
set_help(
"Input image height.");
49 GraphVDSRExample(
const GraphVDSRExample &) =
delete;
50 GraphVDSRExample &operator=(
const GraphVDSRExample &) =
delete;
51 ~GraphVDSRExample()
override =
default;
52 bool do_setup(
int argc,
char **argv)
override 55 cmd_parser.parse(argc, argv);
56 cmd_parser.validate();
62 if(common_params.help)
64 cmd_parser.print_help(argv[0]);
69 const unsigned int image_width = model_input_width->value();
70 const unsigned int image_height = model_input_height->
value();
73 std::cout << common_params << std::endl;
74 std::cout <<
"Image width: " << image_width << std::endl;
75 std::cout <<
"Image height: " << image_height << std::endl;
78 const std::string data_path = common_params.data_path;
79 const std::string model_path =
"/cnn_data/vdsr_model/";
82 std::unique_ptr<IPreprocessor> preprocessor = std::make_unique<TFPreproccessor>();
92 graph << common_params.target
93 << common_params.fast_math_hint
110 for(
unsigned int i = 1; i < 19; ++i)
135 <<
OutputLayer(std::make_unique<DummyAccessor>(0));
140 config.
use_tuner = common_params.enable_tuner;
143 config.
mlgo_file = common_params.mlgo_file;
146 graph.finalize(common_params.target, config);
150 void do_run()
override 177 int main(
int argc,
char **argv)
179 return arm_compute::utils::run_example<GraphVDSRExample>(argc, argv);
Graph configuration structure Device target types.
CLTunerMode tuner_mode
Tuner mode to be used by the CL tuner.
std::unique_ptr< graph::ITensorAccessor > get_input_accessor(const arm_compute::utils::CommonGraphParams &graph_parameters, std::unique_ptr< IPreprocessor > preprocessor=nullptr, bool bgr=true)
Generates appropriate input accessor according to the specified graph parameters. ...
bool convert_to_uint8
Convert graph to a synthetic uint8 graph.
std::string to_string(T &&value)
Convert integer and float values to string.
void consume_common_graph_parameters(CommonGraphValidateOptions &options, CommonParams &common_params)
Consumes the consume_common_graph_parameters graph options and creates a structure containing any inf...
Includes all the Graph headers at once.
Common command line options used to configure the graph examples.
Class to parse command line arguments.
Activation Layer Information class.
Copyright (c) 2017-2021 Arm Limited.
std::string mlgo_file
Filename to load MLGO heuristics from.
std::string tuner_file
File to load/store tuning values from.
Quantization information.
int main(int argc, char **argv)
Main program for VGG-based VDSR.
quantized, asymmetric fixed-point 8-bit number unsigned
Padding and stride information class.
const T & value() const
Get the option value.
TensorDescriptor & set_quantization_info(QuantizationInfo tensor_quant_info)
Sets tensor descriptor quantization info.
Num samples, channels, height, width.
TensorShape permute_shape(TensorShape tensor_shape, DataLayout in_data_layout, DataLayout out_data_layout)
Permutes a given tensor shape given the input and output data layout.
TensorDescriptor & set_layout(DataLayout data_layout)
Sets tensor descriptor data layout.
Structure holding all the common graph parameters.
bool use_tuner
Use a tuner in tunable backends.
std::unique_ptr< graph::ITensorAccessor > get_weights_accessor(const std::string &path, const std::string &data_file, DataLayout file_layout=DataLayout::NCHW)
Generates appropriate weights accessor according to the specified path.
int num_threads
Number of threads to use (thread capable backends), if 0 the backend will auto-initialize, if -1 the backend will stay as it is.
Stream frontend class to construct simple graphs in a stream fashion.
DataLayout
[DataLayout enum definition]
ILayer & set_name(std::string name)
Sets the name of the layer.
void set_help(std::string help)
Set the help message for the option.