当我尝试将yolov4权重转换为tflite时,出现ValueError。我无法解决这个问题并找到合适的答案。这个权重文件可以在PycharmProject.py文件中找到,我可以在对象检测代码中找到它。但在转换为tflite的过程中,出现了ValueError。
已使用的GitHub链接:tensorflow_yolov4_tflitetensorflow-颜色4
我引用了这个StackOverflow问答;A、 但它不起作用。无法将YOLOv4转换为tflite
这是错误代码。
(convert_tflite.py和convert.py(到pb(也不起作用。(
C: \Users\HANSUHO\GitClone \totflite>python save_model.py--权重/data/showes_yolov4-custom_final.weights--输出/checkpoints/yolov4-416--input_size 416--型号yolov4--框架tflite
您会检查输入/输出大小吗?4565322和(1024512,3,3(的大小不同。
https://github.com/hunglc007/tensorflow-yolov4-tflite/issues/147#issuecomment-666736983
检查这个,它将解决问题。
基本上,您正在克隆根repo以运行此文件,同时保持core/config.py中的路径不变。