您好!
我想从 Model Zoo 中训练(并编译) TFLite 对象检测模型。 Model Composer 和 Model Maker 似乎都只支持 Yolox 模型,即 ONNX。 那么、如何训练和编译 TFLite 对象检测模型? 我能否对 Model Maker 脚本进行一些修改、以便它们可以正常工作? 或者我是否必须使用其他一些工具来促进这一过程?
祝你一切顺利。
马格努斯·盖德达
This thread has been locked.
If you have a related question, please click the "Ask a related question" button in the top right corner. The newly created question will be automatically linked to this question.
您好!
我想从 Model Zoo 中训练(并编译) TFLite 对象检测模型。 Model Composer 和 Model Maker 似乎都只支持 Yolox 模型,即 ONNX。 那么、如何训练和编译 TFLite 对象检测模型? 我能否对 Model Maker 脚本进行一些修改、以便它们可以正常工作? 或者我是否必须使用其他一些工具来促进这一过程?
祝你一切顺利。
马格努斯·盖德达
尊敬的 Magnus:
我们的培训框架在 编译前都使用 TORCH/TORCHVISION 和 ONNX EXPORT。 我们的 Model Zoo 中的 TFL 模型尚未在任何工具中进行过重新训练,因此我不相信我们有任何 重新训练 tensorflow 模型的示例代码。
根据 Model Zoo 和 TF 模型的子目录,这些模型似乎已直接从 tensorflow 模型存储库中下载(并可能从 TF 转换为 TF-lite )。 Model Zoo 中的目录具有许可证文件(例如 OD tF1-models)、指示这些模型所涵盖的源代码和许可证。
我建议使用 tensorflow 的实施和工具来对模型进行再培训。 只要它们出来的架构与我们在 Model Zoo 中的版本相同、使用 edgeai-tidl-tools 进行编译就应该很简单(如果不是、请告诉我)。 MODEL Zoo 最近进行了更新、提供了 param.yAML 文件 (示例)、这些文件是我们的其他工具(如 edgeai-GST-APPS)所需要的、用于描述进一步的模型详细信息。
Br、
雷塞