20.08
|
#include "armnnTfLiteParser/ITfLiteParser.hpp"
#include "NMS.hpp"
#include <stb/stb_image.h>
#include <armnn/INetwork.hpp>
#include <armnn/IRuntime.hpp>
#include <armnn/Logging.hpp>
#include <armnn/utility/IgnoreUnused.hpp>
#include <cxxopts/cxxopts.hpp>
#include <ghc/filesystem.hpp>
#include <chrono>
#include <fstream>
#include <iostream>
#include <stdlib.h>
Go to the source code of this file.
Macros | |
#define | CHECK_OK(v) |
Functions | |
template<typename TContainer > | |
armnn::InputTensors | MakeInputTensors (const std::vector< armnn::BindingPointInfo > &inputBindings, const std::vector< std::reference_wrapper< TContainer >> &inputDataContainers) |
template<typename TContainer > | |
armnn::OutputTensors | MakeOutputTensors (const std::vector< armnn::BindingPointInfo > &outputBindings, const std::vector< std::reference_wrapper< TContainer >> &outputDataContainers) |
int | LoadModel (const char *filename, ITfLiteParser &parser, IRuntime &runtime, NetworkId &networkId, const std::vector< BackendId > &backendPreferences, bool enableImport=false) |
std::vector< float > | LoadImage (const char *filename) |
bool | ValidateFilePath (std::string &file) |
void | CheckAccuracy (std::vector< float > *toDetector0, std::vector< float > *toDetector1, std::vector< float > *toDetector2, std::vector< float > *detectorOutput, const std::vector< yolov3::Detection > &nmsOut, const std::vector< std::string > &filePaths) |
int | main (int argc, char *argv[]) |
#define CHECK_OK | ( | v | ) |
Definition at line 34 of file TfLiteYoloV3Big-Armnn.cpp.
Referenced by main().
void CheckAccuracy | ( | std::vector< float > * | toDetector0, |
std::vector< float > * | toDetector1, | ||
std::vector< float > * | toDetector2, | ||
std::vector< float > * | detectorOutput, | ||
const std::vector< yolov3::Detection > & | nmsOut, | ||
const std::vector< std::string > & | filePaths | ||
) |
Definition at line 203 of file TfLiteYoloV3Big-Armnn.cpp.
References ARMNN_LOG, yolov3::compare_detection(), armnn::error, armnn::info, and ValidateFilePath().
std::vector<float> LoadImage | ( | const char * | filename | ) |
Definition at line 146 of file TfLiteYoloV3Big-Armnn.cpp.
References ARMNN_LOG, and armnn::error.
Referenced by main().
int LoadModel | ( | const char * | filename, |
ITfLiteParser & | parser, | ||
IRuntime & | runtime, | ||
NetworkId & | networkId, | ||
const std::vector< BackendId > & | backendPreferences, | ||
bool | enableImport = false |
||
) |
Definition at line 100 of file TfLiteYoloV3Big-Armnn.cpp.
References ARMNN_LOG, and armnn::error.
Referenced by main().
int main | ( | int | argc, |
char * | argv[] | ||
) |
Definition at line 430 of file TfLiteYoloV3Big-Armnn.cpp.
References ARMNN_LOG, CHECK_OK, ITfLiteParser::Create(), armnn::fatal, armnn::info, LoadImage(), LoadModel(), armnn::SetAllLoggingSinks(), and armnn::SetLogFilter().
|
inline |
Definition at line 50 of file TfLiteYoloV3Big-Armnn.cpp.
Referenced by InferenceModel< IParser, TDataType >::GetAllQuantizationParams(), and InferenceModel< IParser, TDataType >::Run().
|
inline |
Definition at line 74 of file TfLiteYoloV3Big-Armnn.cpp.
Referenced by InferenceModel< IParser, TDataType >::GetAllQuantizationParams(), and InferenceModel< IParser, TDataType >::Run().
bool ValidateFilePath | ( | std::string & | file | ) |