Преобразовать пользовательскую модель Yolox (.pth) в модель Tflite ⇐ Python

[Anonymous]

Я обучен yolox, используя это репо в пользовательском наборе данных с 8 классами. Окончательным выводом является файл .pth , который мне нужно преобразовать в формат .tflite , но я полностью застрял. Windows. (То же самое относится и к любому другому коллабу, который использует ai-edge-troch < /code>)
Ошибка считывает: < /li>
< /ol>
/usr/local/lib/python3.11/dist-packages/tensorflow/python/framework/load_library.py in load_library(library_location)
149
150 for lib in kernel_libraries:
--> 151 py_tf.TF_LoadLibrary(lib)
152
153 else:

NotFoundError: /usr/local/lib/python3.11/dist-packages/tensorflow/core/kernels/libtfkernel_sobol_op.so: undefined symbol: _ZN10tensorflow15TensorShapeBaseINS_11TensorShapeEEC2EN4absl12lts_202308024SpanIKlEE
< /code>

• через onnx < /code> - я попытался следовать этому учебному пособию, но код учебного пособия сбивается с ошибкой < /li>
  < /ol>
  import onnx_tf
  [...many lines later...]
  ModuleNotFoundError: No module named 'keras.src.engine'
  < /code>
  , который я еще не смог решить из -за проблем с зависимостью. < /p>
  Так что мои вопросы: < /p>
  
  Любые идеи о том, как исправить вышеупомянутый коллаб? Кажется, есть только так много способов преобразования модельного файла, и все это, кажется, какой -то тупик.

Подробнее здесь: https://stackoverflow.com/questions/797 ... lite-model