将自定义 yolo 权重转换为 tflite？

Question

我已经使用 darknet tiny cfg 在 yolov4 上训练了一些自定义数据集。

现在我有三个文件：

classes.names  
yolov4-tiny-custom.cfg
yolov4-tiny-custom.weights

为了对 flutter 应用程序实施 yolo 对象检测，我需要将这些文件转换为：

 .tflite and .txt

我的总数类是 54

如何转换它们？

Answer 1

您可以从 yolo darknet weights 到 savedmodel pb 再到 tflite。生成保存模型后，然后使用 TFLiteConverter 转换为 tflite，

我已将自定义 yolov4 微型模型部署到 android，并带有下面的回购协议。它可能对您的情况有所帮助。

这个仓库有一段时间没有更新了。已对 coco.names 进行硬编码，因此请修改 config.py 代码以包含您的 classes.names。这是修复问题的相关问题，

cannot reshape array of size 1665179 into shape (512,512,3,3)

Convert custom yolo weights to tflite?