在linux下运行yolov5进行车流量识别,出现报错
from unbox_yolov5_deepsort_counting-main/models/common.py", line 13, in
from torch.cuda import amp
ImportError: cannot import name 'amp'
看看你的 pyTorch 版本,以及amp库是否安装了。
望采纳:
可能你的 PyTorch 版本太低,我查找了amp是PyTorch 1.6.0 引入的,你需要升级。
使用该命令行升级:
pip install torch==1.6.0 torchvision==0.7.0 -f https://download.pytorch.org/whl/cu101/torch_stable.html
不知道你这个问题是否已经解决, 如果还没有解决的话:由于缺少具体的错误信息和出错环节,无法确定在哪个环节出现了报错。但从参考资料中可以看到代码中运用了yolov5和deepsort两种算法,因此出错的原因可能与这两个算法相关。建议可先检查安装的依赖环境是否正确,确保pytorch、python和cuda的版本正确。同时,查看代码中涉及到yolov5和deepsort的部分,逐一检查是否有语法错误、逻辑问题、依赖缺失等。如有错误信息可提供,可以更好地帮助解决问题。
如果问题难以解决,也可以尝试到相关社区和论坛上求助,并提供详细的错误信息和出错环节。