ONNX -Modellkonvertierung in TF, dann TFLite -FehlerPython

Python-Programme
Anonymous
 ONNX -Modellkonvertierung in TF, dann TFLite -Fehler

Post by Anonymous »

Ich habe versucht, dieses ONNX -Modell in den Tensorflow umzuwandeln, damit ich es in Tflite umwandeln kann, das ich dann in meiner Flutter -App verwenden werde < /p>
import onnx

from onnx_tf.backend import prepare
import tensorflow

onnx_model = onnx.load('vit_trashnet.onnx')
tf_rep = prepare(onnx_model)
tf_rep.export_graph('vit_trashnet_tf')
< /code>
Aber ich erhalte diesen Fehler immer wieder < /p>
AttributeError: module 'keras._tf_keras.keras' has no attribute '__internal__'
< /code>
Nicht nur dieser Fehler, sondern auch unterschiedliche Fehler, abhängig von der Version der Abhängigkeit, die ich fixiere. Gibt es eine Problemumgehung, die ich versuche, es in Tflite umzuwandeln, da die Flutter -App aufgrund der Bibliothek in Tflite -Modellen viel bequem ist (zumindest weiß ich das). Außerdem beträgt die Größe des Modells 340 MB+- was ein bisschen groß ist, wenn ich es jemals in meiner App verwenden werde, da ich versuchen möchte, das Modell im Offline-Modus oder was auch immer verwendbar zu machen. Der Ordner hat das RAW VIT-Modell und außerhalb des Ordners hat die OnNX-Datei und die vereinfachte.>

Quick Reply

Change Text Case: 
   
  • Similar Topics
    Replies
    Views
    Last post