Yolo-V3 训练模型可以转换为 TensorFlow 模型吗?

时间:2021-07-04 07:04:11

标签: python tensorflow machine-learning computer-vision yolo

我已经在 yolo-v3 中训练了我的门模型,但现在我需要在 TensorFlow-Lite 中使用它。但是,我面临一个问题,即如果我想为 tensorflow 训练我的模型,我需要“.csv”或“.xml”中的注释文件,但我拥有的是“* 。文本”。我确实找到了一个软件,可以通过在图片中绘制矩形来手动创建注释文件,但由于时间不足,我无法对数千张图像执行此操作。 谁能指导我如何处理这种情况?

我点击了以下链接,但结果模型不起作用。 https://medium.com/analytics-vidhya/yolov3-to-tensorflow-lite-conversion-4602cec5c239

1 个答案:

答案 0 :(得分:0)

我认为在您的数据上训练 tensorflow 实现会很好,然后将 tensrflow 模型转换为 tflite 应该很容易 这是 tf 中的 yolov3 :https://github.com/YunYang1994/tensorflow-yolov3 然后使用官方tensorflow代码转换为tflite:https://www.tensorflow.org/lite/convert