自定义YOLOX模型(.pth)转量化.tflite模型遇阻求助

我基于YOLOX仓库在8类自定义数据集上训练得到了.pth模型文件，现在需要将其转换为量化.tflite格式，但尝试多种方法后均陷入困境。

已尝试的方法 #

• 使用ai-edge-torch工具，但依赖的ai-edge-litert-nightly库仅支持Linux系统，我使用Windows无法安装
• 使用ai-edge-torch官方提供的Colab运行环境，但执行import ai_edge_torch时崩溃，错误信息如下：

/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

• 通过ONNX中转转换，参考相关教程操作时出现依赖错误：

import onnx_tf
[...many lines later...]
ModuleNotFoundError: No module named 'keras.src.engine'

该依赖问题至今无法解决。

我的问题 #

1. 如何修复上述Colab环境中的崩溃错误？
2. 有没有可行的YOLOX转量化TFLite的方案？目前尝试的方法都走不通。

内容的提问来源于stack exchange，提问作者Crazy Engineer