admin 管理员组

文章数量: 1087748

错误解决 tensor.detach().numpy() instead.

yolov7-deepsort运行deepsort时候，出现如下错误

RuntimeError: Can't call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead.
 

解决方法：

det_results = det_results.cpu().numpy()

替换为如下代码

if torch.cuda.is_available():

        det_results = det_results.cpu().detach().numpy()

else:

        det_results = det_results.cpu().numpy()

本文标签： 错误解决 tensordetach()numpy() instead

版权声明：本文标题：错误解决 tensor.detach().numpy() instead. 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.roclinux.cn/p/1699085321a327300.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。