解决yolov3_object_tracking中loop of ufunc does not support argument 0 of type Tensor which has问题

最新推荐文章于 2024-11-19 17:28:36 发布

北风催呀催

最新推荐文章于 2024-11-19 17:28:36 发布

阅读量8.8k

点赞数 1

分类专栏：深度学习文章标签：深度学习

本文链接：https://blog.csdn.net/qq_26078693/article/details/105888228

版权

深度学习专栏收录该内容

3 篇文章 0 订阅

订阅专栏

在使用keras-yolov3 + Kalman-Filter 进行人体多目标追踪（含代码）做实验时遇到问题

TypeError: loop of ufunc does not support argument 0 of type Tensor which has no callable arctan method

报错行

self.u = np.round(np.dot(self.F, self.u))
＃或者
self.u = np.round(self.u + np.dot(K, (self.b - np.dot(self.A, self.u))))

断点可以看到传入的参数self.b为Tensor格式，其他都是ndarray。

这是由于我是用了tensorflow下的yolov3，使用了cuda。得到的box等参数为Tensor格式。

因此在计算中心点的函数

calc_center(bboxes, cls_ids, cls_confs, score_limit=0.5)

传递参数时，转换为array格式即可：

转换方式为：

直接从 cuda 中取数据，如 
var_tensor.cuda().data.cpu().numpy()

例如：在参数转换为列表前转换：

bboxes.append([xyxy[0].cuda().data.cpu().numpy(),
                xyxy[1].cuda().data.cpu().numpy(),
                xyxy[2].cuda().data.cpu().numpy(),
                xyxy[3].cuda().data.cpu().numpy()])
cls_confs.append(conf.cuda().data.cpu().numpy())
cls_ids.append(cls.cuda().data.cpu().numpy())