尝试训练用于媒体管道手势识别的自定义模型时,张量形状不兼容

问题描述 投票:0回答:1

我正在尝试在 Google Colab 上训练一个自定义模型,用于 mediapipe 的手势识别在此处找到

我能够上传、解压和加载数据集,但是当训练模型的单元运行时,它立即抛出错误:

ValueError:尝试恢复形状为(21)且名称为custom_gesture_recognizer_out/bias:0的变量时收到形状为(5)不兼容的张量。

Google Colab 中的所有代码均未更改,除了包含我的数据的文件夹的名称。

我试图找出如何编辑现有的单元格,但网上基本上没有关于更改此合作实验室的张量参数的信息。我也找不到任何关于 mediapipe 手势识别的 shape/input_shape/tensor-shape 参考。

我担心这与我扩大潜在状态的数量有关;最初它是 4(无,大鹏,纸,剪刀),但我给了它 20 + 无,形成 21(形状不匹配中引用的数字),但在这种情况下,我不知道 5 来自哪里。

任何帮助将不胜感激。

答案:Google Colab 缓存了之前训练周期的信息。在使用新数据重新训练之前,请确保清除缓存并硬重置 colab(与内核断开连接并创建新连接)

tensorflow tensor mediapipe gesture-recognition
1个回答
0
投票

我遇到了同样的问题,尝试运行相同的代码,但不是在 Colab 中。我正在尝试重新创建相同的代码,但包含 23 个类,并且我得到以下信息: “ValueError:尝试恢复形状为 (23,) 且名称为 custom_gesture_recognizer_out/bias:0 的变量时收到形状为 (5,) 不兼容的张量。”

有什么想法吗?

© www.soinside.com 2019 - 2024. All rights reserved.