RuntimeError: Class values must be smaller than num_classes

在使用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即

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值