UserWarning: No NMS is available. Please upgrade torchvision to 0.3.0+

于 2024-12-16 22:01:08 发布
阅读量431 收藏
点赞数 4
文章标签： 人工智能 目标跟踪 计算机视觉
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.youkuaiyun.com/lijungong/article/details/144518544
版权

最近在使用SSD进行目标检测对比实验时碰到的错误

UserWarning: No NMS is available. Please upgrade torchvision to 0.3.0+
  warnings.warn('No NMS is available. Please upgrade torchvision to 0.3.0+')

后面查看torchvision的版本是0.17.0，查看代码发现torchvision.__version__ >= '0.3.0'（ssd\utils\nms.py第七行）是直接进行字符串比较，这时0.17.0和0.3.0进行对比是一个字符一个字符进行对比，导致比较到1和3时，0.17.0是小于0.3.0的，所以当你的torchvision的版本是0.1x.0时，把torchvision.__version__ >= '0.3.0'修改成torchvision.__version__ >= '0.1y.0'，其中y<x，比如我的torchvision的版本是0.17.0，我修改成 >='0.11.0'、>='0.12.0'、>='0.16.0'等都可以

001abbc
关注
torchvision nms
jacke121的专栏
06-21 6173
if __name__ == '__main__': import torch a=torch.Tensor([[1,1,2,2],[1,1,3.100001,3],[1,1,3.1,3]]) b=torch.Tensor([[0.9],[0.98],[0.980005]]) from torchvision.ops import nms ccc=nms(a,b,0.4) print(ccc) print(a[ccc])
模型训练-SSD模型
最新发布
qq_69854365的博客
10-08 793
SSD（Single Shot MultiBox Detector）是一种用于目标检测的深度学习模型，它是一种快速、高效的检测算法，广泛应用于计算机视觉领域。SSD的主要特点是可以在单个前向传递中同时进行目标检测和定位，而无需额外的复杂步骤或多次前向传递。
报错 UserWarning: No NMS is available. Please upgrade torchvision to 0.3.0+
m0_46256682的博客
08-17 1180
原因是python比较版本号的时候，按照字符下标顺序比较，如：0.13.1和0.3.0，比较的时候是13的1和3进行比较，就会报错。我是显示正常版本号，错误的话会显示false。显示false的情况这里不讨论。先检查pytorch 和 torchvision 是否正确安装。...
遇到问题UserWarning: No NMS is available.
dear_queen的博客
01-21 2578
UserWarning: No NMS is available. Please upgrade torchvision to 0.3.0+ warnings.warn(‘No NMS is available. Please upgrade torchvision to 0.3.0+’) 好好的环境布吉岛咋的就坏了，遇到了这个问题。 当然，实在不知道怎么解决就重新安装一个新的虚拟环境就好了，但是这个环境已经安了很多包了，直接丢掉太浪费了。 而且我还提示说cuda也不支持了，就很奇怪。 首先测试下tor.
解决torch.cuda.is_available()返回结果为False
weixin_41803482的博客
10-22 3万+
装了好几天的环境，死活用不了GPU torch.cuda.is_available()返回结果一遍一遍的都是False，心态都崩了！！！ 一开始以为是nvidia驱动的问题，然后看了几篇帖子，总结一下我的错误 一、cuda版本问题 cuda是向下兼容的，一开始我用的cuda10.1的docker镜像，本机是cuda10.0的，所以返回False。要确保新建的环境的cuda版本要比本地的基础环境低！！！ 二、pytorch和cuda版本对应问题 装的torch==1.4.0版本的，用print(torch.
解决安装torchvision自动更新torch到最新版本
lxx4610的博客
09-19 2万+
不使用pip install torchvision=0.4.0 使用pip install --no-deps torchvision=0.4.0w 完美解决安装完torchvision之后继续更新torch到1.2.0版本（目前最新版）。
torchvision 安装出现错误
小花生的博客
05-13 1万+
(base) C:\Users\pxsj_admin>pip install torchvision Collecting torchvision Using cached https://files.pythonhosted.org/packages/fb/01/03fd7e503c16b3dc262483e5555ad40974ab5da8b9879e164b56c1f4ef6f/tor...
PyTorch编写代码遇到的问题
二分掌柜的
03-03 1万+
训练人脸检测mtcnn遇到的问题 flyfish no module named xxx xxx为自定义文件夹的名字 因为搜索不到，所以将当前路径加入到包的搜索目录 解决方法： import sys sys.path.append(’…’) #将上层目录加入到搜索路径中 sys.path.append(’/home/xxx’) # 绝对路径 import os sys.path.append(os...
UserWarning: No NMS is available. Please upgrade torchvision to 0.3.0+ warnings.warn(&#39;No NMS is available. Please upgrade torchvision to 0.3.0+&#39;)
05-26
NMS（非极大抑制）是一种常用的目标检测算法，需要使用 torchvision 库的最新版本才能正常使用。你需要升级 torchvision 到 0.3.0 或更高版本来避免这个警告。你可以使用以下命令来升级 torchvision： ```python !...
torchvision报错：UserWarning: Failed to load image Python extension: Could not find module
weixin_43462804的博客
01-26 3万+
torchvision报错：UserWarning: Failed to load image Python extension: Could not find module 原因就一个，torchvision版本过高，怀疑是新版torchvision自己的锅。刚开始按照官网 pip3 install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio===0.10.1+cu102 -f https://download.pytorch.org
RuntimeError: No such operator torchvision::nms问题解决
yrwang_xd的博客
05-05 1万+
作者今天在跑之前写过的一个算法时遇到了这个问题。torch和torchvision版本和之前用的时侯没有变化，torch==1.4.0+cu100 torchvision==0.5.0+cu100。并且 import torch torch.ops.torchvision.nms 输出了 RuntimeError: No such operator torchvision::nms 问题的...
亲测No such operator torchvision::nms解决办法。
u012833399的博客
10-26 3787
No such operator torchvision::nms 网上有关这个问题的答案一直是版本不对应的问题，经过两天的时间查找最终解决了，和版本的关系部分，在生成的spec文件中的对应位置加上如下内容便可以解决。 datas=[('/root/miniconda3/lib/python3.6/site-packages/torch',"torch"), ('/root/miniconda3/lib/python3.6/site-packages/torchvision',"torchvision"),
RuntimeError: No such operator torchvision::nms问题解决方法
qq_41590635的博客
01-08 1万+
RuntimeError: No such operator torchvision::nms 问题解决方法： 最近在学习YoloV5，跑了代码，训练的时候没有问题，在测试的时候报错如下： 网上搜了一下，发现很多人遇到过这个问题，尤其是在跑Yolov4和Yolov5等对pytorch版本要求比较高的代码。但是解决方案说的大都是pytorch和torchvision的版本不匹配，让升级torchvision版本，或者重新安装pytorch。 但是因为pytorch较大，每次安装比较麻烦，且另一种简单升级to
anaconda安装torch_如何解决在cuda上安装torch后torch.cuda.is_available()返回False
weixin_39989688的博客
11-20 1664
课题组的服务器最近又炸了，不得已换一台稳定点的服务器，让师弟开了账号开始部署自己的代码环境。装了anaconda之后，开始安装用的最多的torch。初安装：使用清华镜像下载安装torch、torchvision，代码如下conda install pytorch torchvision cudatoolkit=10.1 -c pytorch中间遇到了HTTP ERROR的问题，设置了几条清华镜像，...
pytorch使用过程中遇到的警告
legalhighhigh的博客
11-04 1万+
1、UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use tensor.item() to convert a 0-dim tensor to a Python number #原语句： train_loss+=loss.data[0] #修改后： train_lo...
torch.cuda.is_available（）返回False解决方案
热门推荐
一一
03-25 9万+
在配置完环境以后运行程序，程序正常运行但是GPU没有任务，此时判断程序没有在GPU上运行。调试步骤： 先查看cuda安装有没有问题：nvcc -V 可以看到cuda驱动等安装没有问题。 进程序去看.cuda()或者.todevice之类代码，看程序确实放在GPU上了，故排除 但是在查看代码是看到这里是一个and，参数args....
PyTorch专栏（八）：微调基于torchvision 0.3的目标检测模型
TensorFlowNews
09-14 2797
专栏目录： 第一章：PyTorch之简介与下载 PyTorch简介 PyTorch环境搭建 第二章：PyTorch之60分钟入门 PyTorch入门 PyTorch自动微分 PyTorch神经网络 PyTorch图像分类器 PyTorch数据并行处理 第三章：PyTorch之入门强化 数据加载和处理 Py...
安装win10+Anaconda3+cuda9.2+Cudnn+pytorch1.2.0+torchvision0.2.0（遇到的问题）
yayadazui的博客
08-19 1553
网上安装pytorch的教程太多太多了，但是针对不同的版本会出现不同的问题，这里我只想提醒还没有安装或者没有配置成功的小伙伴，一定要注意版本兼容问题！！！！！ 安装步骤我就不在赘述，这里列出我遇到的问题及解决措施： 1. 安装完pytorch后，测试已经成功，接下来安装torchvision使用pip3 install torchvision出现问题： ...
【python】yolov5的torch与torchvision环境问题
聿默的博客
05-15 8069
1.可用环境 torch==1.7.1+cu110 torchvision==0.8.2+cu110 2.错误环境遇到的问题 （1）情况一 torch==1.7.0+cu110 torchvision==0.8.0 训练yolov5时，训练就会开始报错： return torch.ops.torchvision.nms(boxes, scores, iou_threshold) RuntimeError: Could not run 'torchvision::nms' with argu
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值