label_one_hot = F.one_hot(x.to(torch.int64), 40).permute(0, 3, 1, 2)
在对标签进行one-hot编码时候,出现了错误,报错显示:F.one_hot的class参数必须小于真实的类别数。
我用的NYU-depth v2,设的40类,发现没错呀。
然后去搜了搜发现可能出现的问题:
1:x即
在使用PyTorch的F.one_hot进行标签one-hot编码时,遇到错误提示class参数必须小于真实类别数。通过检查数据类型、标签类别数量,发现图片有9个独特类别,但在标签中有0和40,实际上表示41个类别。将class参数更正为41后,编码成功运行。
label_one_hot = F.one_hot(x.to(torch.int64), 40).permute(0, 3, 1, 2)
在对标签进行one-hot编码时候,出现了错误,报错显示:F.one_hot的class参数必须小于真实的类别数。
我用的NYU-depth v2,设的40类,发现没错呀。
然后去搜了搜发现可能出现的问题:
1:x即
767
1万+
1万+

被折叠的 条评论
为什么被折叠?