从0开始安装并运行yolov5(仅供自己学习使用)

这篇博客详细介绍了如何从零开始安装并运行yolov5模型。首先通过conda创建一个名为yolo的环境,并安装python 3.8。接着,安装pytorch 1.7.1和相关依赖。权重文件需要从yolov5的GitHub releases页面下载并放入weights目录。运行检测代码后,结果将在指定目录下生成。最后,介绍如何将个人数据集转换为VOC格式,修改配置文件进行训练。
摘要由CSDN通过智能技术生成

1.创建环境 conda create -n yolo python = 3.8
2.安装pytorch1.7以上,这边安装了1.7.1版本的
pip install torch1.7.1+cu101 torchvision0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
用下列指令安装其他包

git clone https://github.com/ultralytics/yolov5
pip install -U -r yolov5/requirements.txt

建议一个一个安装,这样保证不会出错。
3.下载权重,运行检测代码
去官网https://github.com/ultralytics/yolov5/releases
下载自己需要的模型权重,放到./weights里面
然后运行下列代码测试

    python detect.py --source data/images --weights yolov5s.pt --conf 0.25

运行成功的话,生成图片放在./runs/detect/exp里面

4.train自己的数据集。
(弄成VOC格式)
首先,自己的训练集格式要有voc的一样
在这里插入图片描述
images里面放自己训练的图片
labels里面放yolo格式的txt文件。
下面附xml格式转化为txt格式代码。

impor
  • 3
    点赞
  • 25
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值