20.02
CaffeInception_BN-Armnn.cpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
#include "../InferenceTest.hpp"
6
#include "../CaffePreprocessor.hpp"
7
#include "
armnnCaffeParser/ICaffeParser.hpp
"
8
9
int
main
(
int
argc,
char
* argv[])
10
{
11
int
retVal = EXIT_FAILURE;
12
try
13
{
14
// Coverity fix: The following code may throw an exception of type std::length_error.
15
std::vector<ImageSet> imageSet =
16
{
17
{
"shark.jpg"
, 3694}
18
};
19
20
using
DataType
= float;
21
using
DatabaseType =
CaffePreprocessor
;
22
using
ParserType =
armnnCaffeParser::ICaffeParser
;
23
using
ModelType =
InferenceModel<ParserType, DataType>
;
24
25
// Coverity fix: ClassifierInferenceTestMain() may throw uncaught exceptions.
26
retVal = armnn::test::ClassifierInferenceTestMain<DatabaseType, ParserType>(
27
argc, argv,
"Inception-BN-batchsize1.caffemodel"
,
true
,
28
"data"
,
"softmax"
, { 0 },
29
[&imageSet](
const
char
* dataDir,
const
ModelType&) {
30
return
DatabaseType(dataDir, 224, 224, imageSet);
31
});
32
}
33
catch
(
const
std::exception& e)
34
{
35
// Coverity fix: BOOST_LOG_TRIVIAL (typically used to report errors) may throw an
36
// exception of type std::length_error.
37
// Using stderr instead in this context as there is no point in nesting try-catch blocks here.
38
std::cerr <<
"WARNING: CaffeInception_BN-Armnn: An error has occurred when running "
39
"the classifier inference tests: "
<< e.what() << std::endl;
40
}
41
return
retVal;
42
}
InferenceModel
Definition:
InferenceModel.hpp:316
armnnCaffeParser::ICaffeParser
Definition:
ICaffeParser.hpp:24
armnn::DataType
DataType
Definition:
Types.hpp:32
main
int main(int argc, char *argv[])
Definition:
CaffeInception_BN-Armnn.cpp:9
CaffePreprocessor
Definition:
CaffePreprocessor.hpp:18
ICaffeParser.hpp
tests
CaffeInception_BN-Armnn
CaffeInception_BN-Armnn.cpp
Generated on Fri Mar 13 2020 16:09:13 for ArmNN by
1.8.13