yolov5初上手

根据深入浅出Yolov5之自有数据集训练超详细教程进行学习过程中,
强烈建议用conda创建新的虚拟环境,重新在有requirements.txt的目录下pip install -r requirements.txt,可以避免很多问题

安装pytorch可能出现的问题。

出现问题1: 在用pip install -r requirements.txt的时候,安装的是最新版本的torch、和torchvision,运行detect.py时报错 ImportError: Failed to load PyTorch C extensions:,检查到可能情况如下:

  1. 本来的python库中已安装numpy,在用requirements.txt安装时又覆盖安装了新的numpy,导致存在多个numpy。
  • 解决办法: pip uninstall numpy,出现询问是否删除,选择 是,此时卸载后只剩一个numpy
  1. 默认安装的最新版torch,但是用不了。
  • 解决办法:卸载torch,torchvision,torchaudio。重新安装旧版本,经试验,torch== 1.9.0 + torchvision ==0.10.0 + torchaudio == 0.9.0 是可以的,对应关系见官方repo
  • 卸载: pip uninstall torch,pip uninstall torchvision,pip uninstall torchaudio,来源博客

出现问题2: ModuleNotFoundError: No module named ‘kiwisolver’

解决办法:卸载kiwi solver,pip uninstall kiwisolver,再重新安装,pip install kiwisolver.

学习过程中注意:

  • main.py中,将阶段一的注释可以不用打开,只要跑阶段二阶段三就可以了。不然阶段一会弹出很多张图片而且linux命令行中退不掉!

猜你喜欢

转载自blog.csdn.net/m0_51711485/article/details/125840571