遇到问题:
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
若无法解决欢迎留言,我看到回复。
本文最后更新于2020年10月29日,已超过 1 年没有更新,如果文章内容或图片资源失效,请留言反馈,我们会及时处理,谢谢!