用于 Tensor Lite 的 .tflite 文件
作者: anh-ntn-amira创建于 2025年8月29日更新于 2025年8月29日
大家好, 我目前正在使用 TensorFlow Lite 将 U²-Net 集成到 Android 项目中。我的主要目标是实现实时背景消除, 其中输出是主体(前景), 背景已被消除或变为透明。 我尝试了将 PyTorch → ONNX → TensorFlow → TFLite 管道转换, 但我遇到了几个问题(量化、转换错误以及在设备上运行时出现意外的内部错误)。 因此, 我谨请求如果有人已经拥有一个运行的 U²-Net .tflite 版本(无论是完整版还是 U²-Netp), 其特点是:
内容来源: xuebinqin/U-2-Net