将自定义yolo权重转换为tflite



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

现在我有三个文件:

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

为了实现yolo对象检测到flutter应用程序,我需要将这些文件转换为:

.tflite and .txt 

我的班级总数是54

如何转换它们?

您可以从yolo暗网权重到savedmodel pb到tflite。生成savedmodel后,使用TFLiteCoverter转换为tflite,

https://www.tensorflow.org/lite/convert

我已经将自定义的yolov4小模型部署到下面的repo安卓系统。这可能对你的情况有所帮助。

https://github.com/hunglc007/tensorflow-yolov4-tflite

此回购在一段时间内没有更新。具有硬编码的coco.names,因此修改config.py代码以包含您的classes.names。这是解决问题的相关问题,

无法将1665179大小的阵列整形为形状(512512,3,3(

相关内容

  • 没有找到相关文章

最新更新