MCPcopy Create free account
hub / github.com/alibaba/MNN / EnumResNetTypeByString

Function EnumResNetTypeByString

benchmark/exprModels/ResNetExpr.hpp:21–33  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

19};
20
21static inline ResNetType EnumResNetTypeByString(const std::string& key) {
22 auto resNetTypeMap = std::map<std::string, ResNetType>({
23 {"18", ResNet18},
24 {"34", ResNet34},
25 {"50", ResNet50},
26 {"101", ResNet101},
27 {"152", ResNet152}});
28 auto resNetTypeIter = resNetTypeMap.find(key);
29 if (resNetTypeIter == resNetTypeMap.end()) {
30 return (ResNetType)(-1);
31 }
32 return resNetTypeIter->second;
33}
34
35MNN::Express::VARP resNetExpr(ResNetType resNetType, int numClass);
36

Callers 1

mainFunction · 0.85

Calls 2

findMethod · 0.45
endMethod · 0.45

Tested by

no test coverage detected