YOLOv5代码复现流程

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"

成功开始运行

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值