赞
踩
1.github下载源码https://github.com/ultralytics/yolov5/blob/master/README.zh-CN.md
2.conda安装虚拟环境conda create --name YOLOv5 python=3.8,虚拟环境名称YOLOv5,python版本3.8
3.将刚刚下载的YOLOv5源码复制到D:\science\anaconda\anacondaData\envs\YOLOv5目录下
4.从D:\science\anaconda\anacondaData\envs\YOLOv5目录下用activate YOLOv5命令进入虚拟环境
5.在虚拟环境下用pip install -r requirements.txt命令安装项目所需环境(后面加上-i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn梯子加快下载速度)
6.安装GPU版本的pytorch:conda install pytorch==1.10.0 torchvision==0.11.0 cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/用这个命令
7.推理:首先使用原来的摄像头推理python detect.py --weights yolov5s.pt --source 0(0代表摄像头)后面还可以 加以下:
img.jpg # image
vid.mp4 # video
screen # screenshot
path/ # directory
list.txt # list of images
list.streams # list of streams
'path/*.jpg' # glob
'https://youtu.be/Zgi9g1ksQHc' # YouTube
'rtsp://example.com/media.mp4' # RTSP, RTMP, HTTP stream
8.保存一个视频到和detect.py同一根目录下后,将其命名为a.mp4,运行python detect.py --weights yolov5s.pt --source a.mp4,运行结果将保存到D:\science\anaconda\anacondaData\envs\YOLOv5\yolov5-7.0\runs\detect\exp4,若当前无runs文件夹,运行成功后会自动创建,刷新即可
9.训练:运行代码python train.py --data coco.yaml --epochs 10 --weights '' --cfg yolov5n.yaml --batch-size 8,但是一直出现错误:
ImportError: Failed to initialize: Bad git executable.
The git executable must be specified in one of the following ways:
- be included in your $PATH
- be set via $GIT_PYTHON_GIT_EXECUTABLE
- explicitly set via git.refresh()
试了添加环境变量和升级 GitPython 库等多种方式还是报一样的错,后在csdn上找到解决办法:
打开train.py文件,在import os代码后,在下一行加入os.environ["GIT_PYTHON_REFRESH"] = "quiet"
‘’‘’
#出错原因:git环境变量设置问题
#简便解决办法:在导入包的上方增加以下代码
#import os
#os.environ["GIT_PYTHON_REFRESH"] = "quiet"
‘’‘’
9.1上述运行成功后开始下载coco数据集,但是又遇到OSError: [WinError 1455] 页面文件太小,无法完成操作。 Error loading "D:\science\anaconda\anacondaData\envs\YOLOv5\lib\site-packages\torch\lib\cudnn_adv_infer64_8.dll" or one of its dependencies.这个问题,在D盘中扩展内存后依然报同样的错误,重启后生效
9.2生效不到一分钟
问题:RuntimeError: DataLoader worker (pid(s) 2700, 4684, 17088, 2484) exited unexpectedly
解决方法:在dataLoader.py函数中修改num_workers=nw为num_workers=0,共修改了两个
9.3修改后运行,
问题:Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.
解决方法:在train.py文件中,在import.os代码下添加os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
成功开始运行
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。