首先就是需要下载Anaconda创建虚拟环境,这是目前最方便的。如何下载具体参照我的另一文章。
然后创建一个虚拟环境
conda create -n py python=3.8
创建成功,进入虚拟环境
conda activate py
下载pytorch,具体下载什么版本可去官网选择,这里我给出我的
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
然后就是慢慢等待,如果觉得下在速度慢,可以选择换源,换得是 conda的源(也不是很慢,最好别换了,想换自己查一下就行,很简单)
下载yolov7
git clone https://github.com/WongKinYiu/yolov7.git
如果这个总是没有反应,直接下载压缩包 https://github.com/WongKinYiu/yolov7.git
下载完成进入文件夹
cd yolov7
安装依赖
pip install -r requirements.txt
然后运行
python detect.py --weights yolov7.pt --conf 0.25 --img-size 640 --source inference/images/horses.jpg
看看是否缺少别的模块,单独pip下载一下就可以,成功则会首先自动下载一个权重文件然后就在文件夹中出现已经预测好的图片。