YOLO 配置,测试与训练
发布日期:2022-01-31 02:37:42 浏览次数:24 分类:技术文章

本文共 1104 字,大约阅读时间需要 3 分钟。

YOLO下载与安装

Linux系统中,一次运行下列命令即可安装

git clone https://github.com/pjreddie/darknet.gitcd darknetmake

注意

若使GPU和opencv, 需要预先配置cuda ,cudnn和opencv, 然后将makefile文件中做出如下改变后,’ make ’

OPENCV=1GPU=1CUDNN=1

检测

首先下载与模型相对应的预训练模型

wget https://pjreddie.com/media/files/yolov3.weights
然后在darknet根目录下,运行如下命令
./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg
注意cfg文件夹下已经定义了许多网络模型,方便调用

训练

VOC数据库

  • 下载voc
    wget https://pjreddie.com/media/files/VOCtrainval_11-May-2012.tar wget https://pjreddie.com/media/files/VOCtrainval_06-Nov-2007.tar wget https://pjreddie.com/media/files/VOCtest_06-Nov-2007.tar tar xf VOCtrainval_11-May-2012.tar tar xf VOCtrainval_06-Nov-2007.tar tar xf VOCtest_06-Nov-2007.tar
  • 生成label
    在script文件夹下运行
    wget https://pjreddie.com/media/files/voc_label.py python voc_label.py
    建立训练数据集
cat 2007_train.txt 2007_val.txt 2012_*.txt > train.txt
  • 修改cfg/voc.data文件
    ` 1 classes= 20
    2 train = /train.txt
    3 valid = 2007_test.txt
    4 names = data/voc.names
    5 backup = backup
  • 下载预训练的模型
    wget https://pjreddie.com/media/files/darknet53.conv.74
  • 最终训练
./darknet detector train cfg/voc.data cfg/yolov3-voc.cfg darknet53.conv.74

转载地址:https://blog.csdn.net/qq_23858101/article/details/84891786 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!

上一篇:Torch dp package
下一篇:OpenCV 窗口显示多幅图片

发表评论

最新留言

初次前来,多多关照!
[***.217.46.12]2024年04月06日 13时51分25秒

关于作者

    喝酒易醉,品茶养心,人生如梦,品茶悟道,何以解忧?唯有杜康!
-- 愿君每日到此一游!

推荐文章

spring boot 与 Ant Design of Vue 实现新增组织(二十四) 2019-04-27
spring boot 与 Ant Design of Vue 实现修改组织(二十五) 2019-04-27
spring boot 与 Ant Design of Vue 实现删除组织(二十六) 2019-04-27
spring boot 与 Ant Design of Vue 实现获取用户列表(二十七) 2019-04-27
spring boot 与 Ant Design of Vue 实现新增用户(二十八) 2019-04-27
spring boot 与 Ant Design of Vue 实现修改用户(二十九) 2019-04-27
spring boot 与 Ant Design of Vue 实现删除用户(三十) 2019-04-27
spring boot 与 Ant Design of Vue 鉴权体系登录的实现(三十一) 2019-04-27
spring boot 与 Ant Design of Vue 鉴权体系获取用户信息的实现(三十二) 2019-04-27
Druid连接池实现自定义场景的多数据库的连接 2019-04-27
CentOs7命令行(静默)的方式安装oracle数据库 2019-04-27
基于VMware安装CentOs7的镜像 2019-04-27
PL/SQL数据库管理工具的使用 2019-04-27
带你玩转属于自己的spring-boot-starter系列(一) 2019-04-27
带你玩转属于自己自己的spring-boot-starter系列(二) 2019-04-27
带你玩转属于自己的spring-boot-starter系列(三) 2019-04-27
基于SnowFlake算法如何让分库分表中不同的ID落在同一个库的算法的实现 2019-04-27
基于springboot的ShardingSphere5.X的分库分表的解决方案之分库解决方案(二) 2019-04-27
基于springboot的ShardingSphere5.X的分库分表的解决方案之分表解决方案(一) 2019-04-27
基于springboot的ShardingSphere5.X的分库分表的解决方案之关联查询解决方案(三) 2019-04-27