20.02
CaffeSqueezeNet1_0-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
using
DataType
= float;
12
using
DatabaseType =
CaffePreprocessor
;
13
using
ParserType =
armnnCaffeParser::ICaffeParser
;
14
using
ModelType =
InferenceModel<ParserType, DataType>
;
15
16
return
armnn::test::ClassifierInferenceTestMain<DatabaseType, ParserType>(
17
argc, argv,
"squeezenet.caffemodel"
,
true
,
18
"input"
,
"prob"
, { 0 },
19
[](
const
char
* dataDir,
const
ModelType &) {
return
CaffePreprocessor
(dataDir); });
20
}
InferenceModel
Definition:
InferenceModel.hpp:316
armnnCaffeParser::ICaffeParser
Definition:
ICaffeParser.hpp:24
armnn::DataType
DataType
Definition:
Types.hpp:32
CaffePreprocessor
Definition:
CaffePreprocessor.hpp:18
ICaffeParser.hpp
main
int main(int argc, char *argv[])
Definition:
CaffeSqueezeNet1_0-Armnn.cpp:9
tests
CaffeSqueezeNet1_0-Armnn
CaffeSqueezeNet1_0-Armnn.cpp
Generated on Fri Mar 13 2020 16:09:13 for ArmNN by
1.8.13