用paddleclas训练时出错了
收藏
ValueError: (InvalidArgument) If Attr(soft_label) == false, the axis dimension of Input(Label) should be 1.
[Hint: Expected labels_dims[axis] == 1UL, but received labels_dims[axis]:3 != 1UL:1.] (at /paddle/paddle/fluid/operators/softmax_with_cross_entropy_op.cc:192)
[operator < softmax_with_cross_entropy > error]
1
收藏
请登录后评论
这里说你标签的维度设置的不对,输入标签的维度为3。因此在计算交叉熵的时候出错了。建议一行行debug下去,看看为什么标签维度不对。
如果要一行行debug,可以在train.py文件中设置一个默认的配置文件,不走命令行启动.
不好意思,打错了,应该是paddleseg
分割标签应该是单通道灰度图,输入3通道的rgb图片所以报错了
所以要把标签图片rgb通道转为单通道吗?
是的
标签维度和模型输出维度对不上
多谢各位大佬解答,已经解决了。 把标签图片rbg通道用脚本换成单通道图片 然后再送去训练就可以了。
你好,请问你的脚本可以发我一份吗?