 

当前位置：马春杰杰  问题解决  代码报错  正文

RuntimeError: No such operator torchvision::nms

2020-09-23 分类：代码报错评论(0) 赞(213)

最新目录‹

遇到问题：
解决：

遇到问题：

Traceback (most recent call last):
  File "efficientdet_test.py", line 79, in <module>
    threshold, iou_threshold)
  File "/home/ubuntu/bigdisk/part1/Yet-Another-EfficientDet-Pytorch/utils/utils.py", line 109, in postprocess
    anchors_nms_idx = batched_nms(transformed_anchors_per, scores_per[:, 0], classes_, iou_threshold=iou_threshold)
  File "/home/ubuntu/.conda/envs/efficientdetpytorch/lib/python3.6/site-packages/torchvision/ops/boxes.py", line 76, in batched_nms
    keep = nms(boxes_for_nms, scores, iou_threshold)
  File "/home/ubuntu/.conda/envs/efficientdetpytorch/lib/python3.6/site-packages/torchvision/ops/boxes.py", line 36, in nms
    return torch.ops.torchvision.nms(boxes, scores, iou_threshold)
  File "/home/ubuntu/.conda/envs/efficientdetpytorch/lib/python3.6/site-packages/torch/_ops.py", line 61, in __getattr__
    op = torch._C._jit_get_operation(qualified_op_name)
RuntimeError: No such operator torchvision::nms

1

2

3

4

5

6

7

8

9

10

11

12

Traceback (most recent call last):

File "efficientdet_test.py", line 79, in <module>

threshold, iou_threshold)

File "/home/ubuntu/bigdisk/part1/Yet-Another-EfficientDet-Pytorch/utils/utils.py", line 109, in postprocess

anchors_nms_idx = batched_nms(transformed_anchors_per, scores_per[:, 0], classes_, iou_threshold=iou_threshold)

File "/home/ubuntu/.conda/envs/efficientdetpytorch/lib/python3.6/site-packages/torchvision/ops/boxes.py", line 76, in batched_nms

keep = nms(boxes_for_nms, scores, iou_threshold)

File "/home/ubuntu/.conda/envs/efficientdetpytorch/lib/python3.6/site-packages/torchvision/ops/boxes.py", line 36, in nms

return torch.ops.torchvision.nms(boxes, scores, iou_threshold)

File "/home/ubuntu/.conda/envs/efficientdetpytorch/lib/python3.6/site-packages/torch/_ops.py", line 61, in __getattr__

op = torch._C._jit_get_operation(qualified_op_name)

RuntimeError: No such operator torchvision::nms

解决：

pip install torch==1.3.0

pip install torchvision==0.4.1

若无法解决欢迎留言，我看到回复。

赞(213) 打赏

版权声明：本文采用知识共享署名4.0国际许可协议 [BY-NC-SA] 进行授权
文章名称：《RuntimeError: No such operator torchvision::nms》
文章链接：https://www.machunjie.com/trouble/code_error/787.html
本站资源仅供个人学习交流，请于下载后24小时内删除，不允许用于商业用途，否则法律问题自行承担。

分享到

马春杰杰 

好好学习，天天向上。加油！！！TG群

相关推荐

评论抢沙发

马春杰杰博客始建于2017年6月14日，博客主要分享网站建设中遇到的问题及解决方案、自己在读书过程中的心得体会及分享一些自觉有意义的音视频内容，记录些生活中的琐事，希望博客能督促怠惰的自己不断学习，不断进步。

精彩直达

友情链接视频解析简历生成表白网页联系我们给我留言

3 + 5 = ?

私密评论

:smile:

:sad:

:cool:

:confused:

:cry:

:eek:

:evil:

:exclaim:

:idea:

:lol:

:mad:

:mrgreen:

:neutral:

:question:

:razz:

:redface:

:rolleyes:

:surprised:

:wink:

:biggrin:

:twisted: