解決'tf.ResizeNearestNeighbor' op is neither a custom op nor a flex op
將tensorflow模型轉換爲TFlite模型:參考個人博客https://panjinquan.blog.csdn.net/article/details/107231435 ,其中可能會出現轉換錯誤:網絡
'tf.ResizeNearestNeighbor' op is neither a custom op nor a flex op
<unknown>:0: error: failed while converting: 'main': Ops that can be supported by the flex runtime (enabled via setting the -emit-select-tf-ops flag): ResizeNearestNeighbor.
測試
自從Tensorflow更新到2.0後,我就發現幾乎每一個版本的都有細微的差別問題,出現這個錯誤,多是因爲的網絡層添加UpSampling2D這些上採樣層,但TFlite是支持UpSampling2D操做的.flex
我能夠很負責任的告訴你,這多是訓練模型的版本,跟你測試的版本不一致致使的,好比當你用TF2.0進行模型訓練時,在TF2.2進行TFlite轉換,就可能出現這個錯誤,你須要保證訓練和轉換TFLite的版本一致便可spa