-

#include "../InferenceTest.hpp"
-#include "../CaffePreprocessor.hpp"
-#include "armnnCaffeParser/ICaffeParser.hpp"
-

-

- - - - -

-Functions
int	main (int argc, char *argv[])

-

Function Documentation

- -

◆ main()

- -

-

- - - - - - - - - - - - - - - - - - -

int main	(	int	argc,
		char *	argv[]
	)

-

- -

Definition at line 9 of file CaffeVGG-Armnn.cpp.

-

 {
     armnn::TensorShape inputTensorShape({ 1, 3, 224, 224 });
     int retVal = EXIT_FAILURE;
     try
     {
         using DataType = float;
         using DatabaseType = CaffePreprocessor;
         using ParserType = armnnCaffeParser::ICaffeParser;
         using ModelType = InferenceModel<ParserType, DataType>;
 
         // Coverity fix: ClassifierInferenceTestMain() may throw uncaught exceptions.
         retVal = armnn::test::ClassifierInferenceTestMain<DatabaseType, ParserType>(
                     argc, argv, "VGG_CNN_S.caffemodel", true,
                     "input", "prob", { 0 },
                     [](const char* dataDir, const ModelType&) {
                         return DatabaseType(dataDir, 224, 224);
                     }, &inputTensorShape);
     }
     catch (const std::exception& e)
     {
         // Coverity fix: BOOST_LOG_TRIVIAL (typically used to report errors) may throw an
         // exception of type std::length_error.
         // Using stderr instead in this context as there is no point in nesting try-catch blocks here.
         std::cerr << "WARNING: CaffeVGG-Armnn: An error has occurred when running "
                      "the classifier inference tests: " << e.what() << std::endl;
     }
     return retVal;
 }

-
-
-
-
-

-

-Functions

Function Documentation

◆ main()