| 19 | }; |
| 20 | |
| 21 | static 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 | |
| 35 | MNN::Express::VARP resNetExpr(ResNetType resNetType, int numClass); |
| 36 |