1

我正在使用 Google Colabs 将 Yolov4 暗网权重转换为 tflite 版本。我使用这个博客来训练我自己的 yolo 检测器,并且我的检测精度达到了可接受的水平。然后我尝试了 github 123上的每个存储库,以将我的自定义 yolov4 权重转换为 tflite 版本,但每次都失败了。我在这里遇到了问题(无法重塑数组),我通过修改 file.names 并调整配置文件中的更改来解决它。最后当我执行这段代码时:
!python convert_tflite.py --weights ./data/yolov4.weights --output ./data/yolov4.tflite

我遇到了这个错误:

ValueError:节点 model_1/batch_normalization/AssignNewValue 的输入 0 从 model_1/batch_normalization/FusedBatchNormV3/ReadVariableOp/resource:0 与预期资源不兼容。

此处讨论了此问题,但我找不到任何解决方案,我正在使用 google colab,这使得使用预安装的库更容易,但转换中使用的库可能存在一些不可比性?
任何帮助,将不胜感激..

4

0 回答 0