聚热网 旅游 outofrange

outofrange

outofrange

python编程语言,使用了pytorch框架的一个读取图像数据集的程序,在运行的时候,其中把图像的标签数字转化为onehot编码的语句报错,该语句如下:

torch.sparse.torch.eye(NUM_CLASSES).index_select(dim=0,index=label)

报错如下:

RuntimeError:indexoutofrangeat..\\aten\\src\\TH/generic/THTensorEvenMoreMath.cpp:193

调试后发现是因为label=-1造成的数组越界,图像的标签数字最小为0,减去1之后-1输入报错语句就越界了,解决办法:label值都加上1就解决了。

outofrange

版权声明:本网站为非赢利网站,作品与素材版权均归作者所有,如内容侵权与违规请发邮件联系,我们将在三个工作日内予以改正,请发送到chengfengseo#gmail.com(#换成@)。

本文链接:https://www.jr-cnc.cn/post/9e61eeec1551e0dd9a04b01d.html

TAG:

相关推荐