tensorflow - “ValueError：无法将大小为 278540 的数组重塑为形状 (256,128,3,3)” 将 YOLOv3 .weights 转换为 .pb

Question

我已经训练了一个 YOLO v3 对象检测模型。为了合并到我的颤振应用程序中，我试图将其转换为 .tflite，需要 .pb 作为中间文件。我尝试过的每个 github 存储库都出现此错误。（下面有几个链接）

错误：ValueError：无法将大小为 278540 的数组重新整形为形状 (256,128,3,3)

以下是我的 classes.names 文件的样子：

我只有2节课。我无法转换。有人可以帮忙吗？

链接到我的权重和配置文件：

• 权重：https ://drive.google.com/file/d/1dqKHqzaqhHuxO5wuW9mgIeCbor9NyTMh/view?usp=sharing

• 配置：https ://drive.google.com/file/d/1_e607nkMTObKlYUCQc8IZtXMsUKJ_aab/view?usp=sharing

我尝试过的一些存储库：

• https://github.com/benjamintanweihao/YOLOv3
• https://github.com/hunglc007/tensorflow-yolov4-tflite
• https://github.com/mystic123/tensorflow-yolo-v3?fbclid=IwAR0ojevKaFH2EIQvxmMfSgLPm9Mz9V_e_5E621BEckgGb-FpvuS7ZLcM7NI
• https://github.com/peace195/tensorflow-lite-YOLOv3
• https://github.com/AlexeyAB/darknet#how-to-train-to-detect-your-custom-objects
• https://github.com/zzh8829/yolov3-tf2
• https://github.com/mystic123/tensorflow-yolo-v3

标签： tensorflowobject-detectiontensorflow-liteyolodarknet

Answer 1

我已经使用这个repo 将我的自定义 YOLO v4 微型模型转换为 TensorFlow。我也面临同样的错误。当我深入研究代码时，默认情况下，代码在./core/utils.py第1./data/classes/coco.names行。78. 我给了我自己的道路obj.names，它奏效了。请给出您的obj.names或 *.names in line no. 78 of./core/utils.py` 的路径。

样本：

with open("path/to/your/obj.names", 'r') as data:

或在线编辑__C.YOLO.CLASSES = "path/to/your/obj.names"。config.py14.