IndexError:too many indices for tensor of dimension 1

2021,12,23
报错语句:

poses=self.pose_model.predic(frame,detected[:,0:4],detected[:,4])

跑的中间报错,输出detected,发现报错帧detected这个张量为空,开始的判断语句写的是:

if len(detected)!=0:

但是当其为空的时候依然执行了该条件句下的代码,所以我认为这个判定语句有问题。

将其改为:

if detected.numel()!=0:

程序跑通。numel()函数返回tensor中元素的个数。

你可能感兴趣的:(敲代码遇到的各种bug,python,pytorch)