其他分享
首页 > 其他分享> > 2021-03-09踩坑 torch.cat

2021-03-09踩坑 torch.cat

作者:互联网

*将两张图像的detections tensor拟合在一起

** 1. 通过循环检测后,两张图像的 tensor 存在 images_detections 中 **
如: [tensor([[1,2], [2,3], [3,4]]), tensor([[1,2], [2,3], [3,4]])]
** 2. 直接对 images_detections 执行自融合 torch.cat(images_detections, 0 ) 即可**
如: tensor([[1,2], [2,3], [3,4], [1,2], [2,3], [3,4]])

标签:03,torch,tensor,09,两张,cat,detections,images
来源: https://blog.csdn.net/weixin_38746088/article/details/114602172