aa23fbb9d3
* Move processing ins and outs into samples/common
- To re-use it in compile_tool and benchmark_app
* Extract common into samples_utils lib
- To re-use it in samples and tools
* Move processLayout into samples_utils
* Clean up
* Enable warning as errors for samples_utils
* Cannot set precision or layout for compiled network
* Remove layout setting for benchmark_app
- It requires further discussion
* Fix compilation of samples
* Fix one more sample
* Fix windows build
* Update README.md
* One more fix for windows
* Fix OpenVino ONNX CI build
* More build fixes
* Move os and vpu common into utils
* Fix build for myriad tools
* Fix windows build
* Export gflags
72 lines
2.6 KiB
C++
72 lines
2.6 KiB
C++
// Copyright (C) 2018-2020 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
#pragma once
|
|
|
|
#include <string>
|
|
#include <map>
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
#include "inference_engine.hpp"
|
|
|
|
InferenceEngine::CNNNetwork readNetwork(const std::string &xmlFileName);
|
|
|
|
bool isFP16(InferenceEngine::Precision precision);
|
|
bool isFP32(InferenceEngine::Precision precision);
|
|
bool isU8(InferenceEngine::Precision precision);
|
|
bool isFloat(InferenceEngine::Precision precision);
|
|
/* Set FP32 to FP16, all others without changes */
|
|
void setPrecisions(const InferenceEngine::CNNNetwork &network);
|
|
|
|
class BitMap {
|
|
private:
|
|
typedef struct {
|
|
unsigned short type; /* Magic identifier */
|
|
unsigned int size; /* File size in bytes */
|
|
unsigned int reserved;
|
|
unsigned int offset; /* Offset to image data, bytes */
|
|
} BmpHeader;
|
|
|
|
typedef struct {
|
|
unsigned int size; /* Header size in bytes */
|
|
int width, height; /* Width and height of image */
|
|
unsigned short planes; /* Number of colour planes */
|
|
unsigned short bits; /* Bits per pixel */
|
|
unsigned int compression; /* Compression type */
|
|
unsigned int imagesize; /* Image size in bytes */
|
|
int xresolution, yresolution; /* Pixels per meter */
|
|
unsigned int ncolours; /* Number of colours */
|
|
unsigned int importantcolours; /* Important colours */
|
|
} BmpInfoHeader;
|
|
|
|
public:
|
|
explicit BitMap(const std::string &filename);
|
|
|
|
~BitMap() = default;
|
|
|
|
size_t _height = 0;
|
|
size_t _width = 0;
|
|
std::shared_ptr<unsigned char> _data;
|
|
|
|
public:
|
|
size_t size() const { return _width * _height * 3; }
|
|
size_t width() const { return _width; }
|
|
size_t height() const { return _height; }
|
|
|
|
std::shared_ptr<unsigned char> getData() {
|
|
return _data;
|
|
}
|
|
};
|
|
|
|
void loadImage(const std::string &imageFilename, InferenceEngine::Blob::Ptr &blob);
|
|
|
|
void printPerformanceCounts(const std::map<std::string, InferenceEngine::InferenceEngineProfileInfo>& perfMap, const std::string report = "per_layer");
|
|
|
|
std::vector<std::string> extractFilesByExtension(const std::string& directory, const std::string& extension);
|
|
std::vector<std::string> extractFilesByExtension(const std::string& directory, const std::string& extension,
|
|
std::size_t max_size);
|
|
|
|
void loadBinaryTensor(const std::string &binaryFileName, InferenceEngine::Blob::Ptr& blob);
|