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+')
时间: 2024-02-01 09:16:59 浏览: 172
这个警告是由于你正在使用 torchvision 的非最新版本导致的。NMS(非极大抑制)是一种常用的目标检测算法,需要使用 torchvision 库的最新版本才能正常使用。你需要升级 torchvision 到 0.3.0 或更高版本来避免这个警告。你可以使用以下命令来升级 torchvision:
```python
!pip install --upgrade torchvision
```
请注意,如果你在使用 conda 管理你的 Python 环境,你需要使用以下命令来升级 torchvision:
```python
!conda install -c pytorch torchvision
```
阅读全文