TypeError: can‘t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to

在mmdetection训练好了maskrcnn做检测,自己写了一个测试脚本,里面调用了mmdetection/mmdet/apis/inference.py里面的show_result_pyplot函数,现在出现如下报错:

TypeError: can‘t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to_第1张图片

经过上网查询了解到这个错误是因为:

 于是寻找CUDA tensor数据格式转numpy时候的代码,于是定位到:

 将这里的return self.numpy()改为return self.cpu().numpy(),大功告成!
 

你可能感兴趣的:(numpy,python,目标检测)