import torchvision问题

最新推荐文章于 2024-06-27 14:02:08 发布

高诗情

最新推荐文章于 2024-06-27 14:02:08 发布

阅读量1.1k

点赞数 2

分类专栏： pytorch 文章标签： torchvision

本文链接：https://blog.csdn.net/qq_33932451/article/details/103204490

版权

pytorch 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

raise NotSupportedError(base.range(), "slicing multiple dimensions at the same time isn't supported yet")
torch.jit.frontend.NotSupportedError: slicing multiple dimensions at the same time isn't supported yet
proposals (Tensor): boxes to be encoded
"""

# perform some unpacking to make it JIT-fusion friendly
wx = weights[0]
wy = weights[1]
ww = weights[2]
wh = weights[3]

proposals_x1 = proposals[:, 0].unsqueeze(1)
~~~~~~~~~ <--- HERE
proposals_y1 = proposals[:, 1].unsqueeze(1)
proposals_x2 = proposals[:, 2].unsqueeze(1)
proposals_y2 = proposals[:, 3].unsqueeze(1)

reference_boxes_x1 = reference_boxes[:, 0].unsqueeze(1)
reference_boxes_y1 = reference_boxes[:, 1].unsqueeze(1)
reference_boxes_x2 = reference_boxes[:, 2].unsqueeze(1)
reference_boxes_y2 = reference_boxes[:, 3].unsqueeze(1)

版本兼容问题：解决办法 pip3 install torchvision==0.2.0（本人用的python版本是3.7）

找了好久才找到答案，希望能够对大家有帮助。