用的这个yolov5模型:
yolov5-master(cow).rar_免费高速下载|百度网盘-分享无限制 (baidu.com)
提取码:2022
电脑须有NVIDIA独立显卡,(绝大多数电脑都符合pytorch要求,不用管CUDA ,CUDNN
假设你已经安装了Anaconda
anaconda prompt打开
输入:
conda create -n Pytorch_GPU python=3.8
安装成功:
输入:
activate Pytorch_GPU
输入:
conda install pytorch==1.8.0 torchvision torchaudio cudatoolkit=10.2
最好安装 1.8.0 的Pytorch版本,电脑都能满足这个版本,否则YOLOv5模型对高版本的torch会报错,我一开始是1.11.2版本的torch,直接去闯就报错了,一查就是版本太高的问题
报错:PackagesNotFoundError: The following packages are not available from current channels:
按照方法换源了也不行,我就直接在pycharm终端安装torch==1.8.0,在官网上下载的
10分钟安装pytorch1.8_Alex抱着爆米花的博客-CSDN博客_pytorch1.8
结果在方法二里面先装的torchversion,torch就下不来了,用豆瓣源,发现pycharm终端-i命令可以换源
pip install "torchvision-0.9.0+cpu-cp39-cp39-win_amd64.whl"-i http://pypi.douban.com/simple --trus
ted-host=pypi.douban.com
发现torch安装好了
如果还是红色的就再重启一下pycharm试试
opencv-python
opencv-contrib-python
pip install -i https://pypi.douban.com/simple --trusted-host pypi.douban.com opencv-python
pip install -i https://pypi.douban.com/simple --trusted-host pypi.douban.com opencv-contrib-python
尝试一下yolov5能跑了不,在pycharm终端里面输入:
python detect.py --source data/images/bus.jpg --weights pretrained/yolov5s.pt
出现哪个包没安装就再用上面那个源或者再搜一下国内其他源的名字去下好所有包
最后再来一次上面那个命令
得到结果!!!!
看到下面那个results saved to 就是结果保存目录
终于把yolo跑通啦,哭哭,我太菜了