我已经使用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(