Tencent / TNN

TNN: developed by Tencent Youtu Lab and Guangying Lab, a uniform deep learning inference framework for mobile、desktop and server. TNN is distinguished by several outstanding features, including its cross-platform capability, high performance, model compression and code pruning. Based on ncnn and Rapidnet, TNN further strengthens the support and performance optimization for mobile devices, and also draws on the advantages of good extensibility and high performance from existed open source efforts. TNN has been deployed in multiple Apps from Tencent, such as Mobile QQ, Weishi, Pitu, etc. Contributions are welcome to work in collaborative with us and make TNN a better framework.
Other
4.38k stars 767 forks source link

TFLite 模型转换出错 #798

Open lhxy1991 opened 3 years ago

lhxy1991 commented 3 years ago

将TFLite模型转化为TNN模型时报错,错误如下: [File /home/xxx/TNN/tools/converter/source/tflite/tflite_converter.cc][Line 112] The TFLiteConverter do not support layer:ArgMax E/tnn: tnn::Status TNN_CONVERTER::TFLite2Tnn::Convert2Tnn(tnn::NetStructure&, tnn::NetResource&) [File /home/xxxx/TNN/tools/converter/source/tflite/tflite_converter.cc][Line 114] The unsupported operator type is:ARG_MAX 请问下这个如何解决呢?

1627180283 commented 3 years ago

应该是模型里有 argmax 这个算子不支持,可以把这个算子去除之后再尝试一下转换

gttiankai commented 3 years ago

@lhxy1991 有模型吗?如果方便的话,上传个模型,我来支持下。