IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Publishing R5 content (#72)

Browse Source 
* Publishing R5 content

* Updated ade revision

* updated readme

* add possibility to build CPU plugin with Intel MKL package
This commit is contained in:
Alexey Suhov
2019-01-21 21:31:31 +03:00
committed by openvino-pushbot
parent fbc7a4a710
commit 9de27f16bc
1616 changed files with 126362 additions and 17710 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
inference-engine/samples/hello_classification/main.cpp
View File

@@ -1,5 +1,4 @@
// Copyright (C) 2018 Intel Corporation
//
// SPDX-License-Identifier: Apache-2.0
//
@@ -9,32 +8,47 @@
#include <string>
#include <cstdlib>
#ifdef UNICODE
#include <tchar.h>
#endif
#include <opencv2/opencv.hpp>
#include <inference_engine.hpp>
using namespace InferenceEngine;
#ifndef UNICODE
#define tcout std::cout
#define _T(STR) STR
#else
#define tcout std::wcout
#endif
#ifndef UNICODE
int main(int argc, char *argv[]) {
#else
int wmain(int argc, wchar_t *argv[]) {
#endif
try {
// ------------------------------ Parsing and validation of input args ---------------------------------
if (argc != 3) {
std::cout << "Usage : ./hello_classification <path_to_model> <path_to_image>" << std::endl;
tcout << _T("Usage : ./hello_classification <path_to_model> <path_to_image>") << std::endl;
return EXIT_FAILURE;
}
const std::string input_model{argv[1]};
const std::string input_image_path{argv[2]};
const file_name_t input_model{argv[1]};
const file_name_t input_image_path{argv[2]};
// -----------------------------------------------------------------------------------------------------
// --------------------------- 1. Load Plugin for inference engine -------------------------------------
PluginDispatcher dispatcher({"../../../lib/intel64", ""});
PluginDispatcher dispatcher({_T("../../../lib/intel64"), _T("")});
InferencePlugin plugin(dispatcher.getSuitablePlugin(TargetDevice::eCPU));
// -----------------------------------------------------------------------------------------------------
// --------------------------- 2. Read IR Generated by ModelOptimizer (.xml and .bin files) ------------
CNNNetReader network_reader;
network_reader.ReadNetwork(input_model);
network_reader.ReadWeights(input_model.substr(0, input_model.size() - 4) + ".bin");
network_reader.ReadNetwork(fileNameToString(input_model));
network_reader.ReadWeights(fileNameToString(input_model).substr(0, input_model.size() - 4) + ".bin");
network_reader.getNetwork().setBatchSize(1);
CNNNetwork network = network_reader.getNetwork();
// -----------------------------------------------------------------------------------------------------
@@ -64,7 +78,7 @@ int main(int argc, char *argv[]) {
// --------------------------- 6. Prepare input --------------------------------------------------------
cv::Mat image = cv::imread(input_image_path);
cv::Mat image = cv::imread(fileNameToString(input_image_path));
/* Resize manually and copy data from the image to the input blob */
Blob::Ptr input = infer_request.GetBlob(input_name);