安装pytorch之前,我们先配置清华镜像,这样下载回快点
在anaconda终端运行如下命令即可:
# 添加清华镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
# 添加pytorch镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# for legacy win-64
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --set show_channel_urls yes
pytorch安装:PyTorch
去掉后面的-c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.3
从GitHub上直接搜索 yolov7 down下来就行。也可以从这里下载:YOLOv7从GitHub上下载下来的代码,根据之前的5版本对比debug了一下,可以测试本地视频了-Python文档类资源-CSDN下载
安装所需的依赖库:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt
控制台运行脚本即可:
Run inference on images, videos, directories, streams, etc.
Usage - sources:
$ python path/to/detect.py --weights yolov5s.pt --source 0 # webcam
img.jpg # image
vid.mp4 # video
path/ # directory
path/*.jpg # glob
'https://youtu.be/Zgi9g1ksQHc' # YouTube
'rtsp://example.com/media.mp4' # RTSP, RTMP, HTTP stream
D:\yolov7-main>python detect.py --weights yolov7.pt --source ./data/image/bus.jpg