ubuntu18安装、测试YOLOV3记录

news2024/11/19 10:30:59

官方教程:

YOLO: Real-Time Object Detection

一、使用预训练模型进行检测

1、安装Darknet:

git clone https://github.com/pjreddie/darknet
cd darknet
make

2、下载预训练权重https://pjreddie.com/media/files/yolov3.weights(打开链接或wget)

https://pjreddie.com/media/files/yolov3.weights

wget https://pjreddie.com/media/files/yolov3.weights

3、运行检测器

./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg

darknet/data文件夹下有还有其他图片可以进行测试。

data/eagle.jpgdata/dog.jpgdata/person.jpg, or data/horses.jpg!

检测命令和下面这条命令是等同的,如果只是想测试一张图片的话,不需要了解。如果想做其他更多的事情要了解该命令:

./darknet detector test cfg/coco.data cfg/yolov3.cfg yolov3.weights data/dog.jpg

4、多个图像

在提供图像命令部分,使用空格,此时将连续测试多个图像,配置和权重加载完成后可以看到如下提示:

./darknet detect cfg/yolov3.cfg yolov3.weights

Loading weights from yolov3.weights...Done!
Enter Image Path:

可以任选以下路径之一输入:

data/eagle.jpgdata/dog.jpgdata/person.jpg, or data/horses.jpg!

Loading weights from yolov3.weights...Done!
Enter Image Path: data/eagle.jpg
data/eagle.jpg: Predicted in 19.587430 seconds.
bird: 99%
Enter Image Path:

然后可以再次输入路径,使用ctrl-c或命令有错误时可以退出程序。

5、改变检测阈值

默认情况下,yolo仅显示检测到的置信度为0.25或者更高的对象,可以通过将-thread <val>标志传递给yolo更改设置,改为0表示显示所有检测。

./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg -thresh 0

6、微型yolov3

yolov3-tiny非常小的模型用于受限环境,使用此模型,下载权重:

wget https://pjreddie.com/media/files/yolov3-tiny.weights

运行检测器

./darknet detect cfg/yolov3-tiny.cfg yolov3-tiny.weights data/dog.jpg

Loading weights from yolov3-tiny.weights...Done!
data/dog.jpg: Predicted in 0.771317 seconds.
dog: 57%
car: 52%
truck: 56%
car: 62%
bicycle: 59%

小的模型训练时间明显提升,本机测试时间从17~19s降到0.77s。

二、摄像头实时检测

1.运行此演示,需要cuda和opencv,安装教程如下:

cuda教程:(真心推荐该教程,按头安利)

深度学习必备NVIDIA三件套环境,大佬同学写了不错的教程,推荐给大家,祝大家科研顺利:

NVIDIA三件套环境配置 - PleasureReceiver

NVIDIA三件套安装问题汇总 - PleasureReceiver

这两篇文章+百度解决所有问题,本人使用该教程多次成功安装三件套。

opencv教程:

在ubuntu16.04下安装opencv3.4.10(超详细测试成功)_再打三年球的博客-CSDN博客_opencv-3.4.10压缩包下载

此方法安装依赖的时候,E: 无法定位软件包 libjasper-dev,使用下面方法的方式安装依赖。

ubuntu安装opencv的正确方法_普通网友的博客-CSDN博客_opencv ubuntu

 2.编译 Darknet with CUDA and OpenCV

(1)with cuda

修改Makefile

GPU=1

默认在0号显卡运行,如果想换卡,添加参数-i, -i <index>,eg:

./darknet -i 1 imagenet test cfg/alexnet.cfg alexnet.weights

如果使用CUDA编译但是又想使用CPU,使用命令 -nogpu,eg:

./darknet -nogpu imagenet test cfg/alexnet.cfg alexnet.weights

(2)with opencv

Darknet默认使用stb_image.h加载图像,想要使用更多格式支持就要改用opencv,使用opencv不需要将图像保存到磁盘就可以查看图像和检测结果。

同理修改Makefile

OPENCV=1

修改完后,需要重新make以下工程,报错:是cudann版本问题,官网有更新对应的文件

./src/convolutional_layer.c:153:13: error: ‘CUDNN_CONVOLUTION_FWD_SPECIFY_WORKSPACE_LIMIT’ undeclared (first use in this function); did you mean ‘CUDNN_CONVOLUTION_FWD_ALGO_DIRECT’?
             CUDNN_CONVOLUTION_FWD_SPECIFY_WORKSPACE_LIMIT,
             ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
             CUDNN_CONVOLUTION_FWD_ALGO_DIRECT
compilation terminated due to -Wfatal-errors.
Makefile:90: recipe for target 'obj/convolutional_layer.o' failed
make: *** [obj/convolutional_layer.o] Error 1

如何解决pjreddie版darknet不能使用cudnn8编译的问题_Arnold-FY-Chen的博客-CSDN博客_./src/convolutional_layer.c:153:13: error: 鈥楥udnn_

可以去官网找到相应文件进行替换:

GitHub - arnoldfychen/darknet: Convolutional Neural Networks

又有new bug:opencv版本问题,本人安装的opencv是3.4.10

./src/image_opencv.cpp:63:20: error: conversion from ‘cv::Mat’ to non-scalar type ‘IplImage {aka _IplImage}’ requested
     IplImage ipl = m;
                    ^
compilation terminated due to -Wfatal-errors.
Makefile:87: recipe for target 'obj/image_opencv.o' failed
make: *** [obj/image_opencv.o] Error 1

DarkNet——OpenCV版本遇到的问题_Captain_zp的博客-CSDN博客

new new bug;

nvcc fatal   : Unsupported gpu architecture 'compute_30'
Makefile:93: recipe for target 'obj/convolutional_kernels.o' failed
make: *** [obj/convolutional_kernels.o] Error 1

这篇cadn教材有提到过这个bug,把Makefile里的配置修改一下

如何解决pjreddie版darknet不能使用cudnn8编译的问题_Arnold-FY-Chen的博客-CSDN博客_./src/convolutional_layer.c:153:13: error: 鈥楥udnn_

参考csdn教程:本教程指出,Makefile文件中的NVCC也要修改,NVCC=/usr/local/cuda-9.0/bin/nvcc,原为 NVCC = nvcc,cuda后面的紧跟自己的cuda版本,如cuda-9.0,cuda-10.0等

Ubuntu18.04下YOLOv3详解——搭建、训练、测试、评估_夜夜coding到天明的博客-CSDN博客_yolo v3 imx8

再次make后成功,然后使用 imtest 例程测试图片加载和显示:

./darknet imtest data/eagle.jpg

3.现在cuda和opencv编译完成,运行命令:

./darknet detector demo cfg/coco.data cfg/yolov3.cfg yolov3.weights

YOLO 将显示当前的 FPS 和预测的类别,以及在其上绘制边界框的图像

 

 如果连接了多个摄像头,可以添加参数-c <num>,选择摄像头,opencv有指令可以查看摄像头编号,笔记本自带摄像头默认是0。

如果 OpenCV 可以读取视频,也可以在视频文件上运行它:

./darknet detector demo cfg/coco.data cfg/yolov3.cfg yolov3.weights <video file>

测试了一下本人手机拍摄的视频.mp4文件,对文件没有做任何处理

 三、VOC数据集上训练

1.下载数据集,VOCdevkit/ 中包含所有 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

2.生成labels文件

wget https://pjreddie.com/media/files/voc_label.py
python voc_label.py

几分钟后完成,此时文件结构如下:

2007_test.txt   VOCdevkit
2007_train.txt  voc_label.py
2007_val.txt    VOCtest_06-Nov-2007.tar
2012_train.txt  VOCtrainval_06-Nov-2007.tar
2012_val.txt    VOCtrainval_11-May-2012.tar

Darknet 需要一个文本文件,其中包含您要训练的所有图像。在此示例中,使用除 2007 测试集之外的所有内容进行训练,以便可以测试模型。运行:

cat 2007_train.txt 2007_val.txt 2012_*.txt > train.txt

4.修改 Pascal 数据的 Cfg

更改 cfg/voc.data 配置文件以指向自己的数据, <path-to-voc>修改为自己的地址

  1 classes= 20
  2 train  = <path-to-voc>/train.txt
  3 valid  = <path-to-voc>2007_test.txt
  4 names = data/voc.names
  5 backup = backup

5.下载预训练的卷积权重

wget https://pjreddie.com/media/files/darknet53.conv.74

6.训练模型

./darknet detector train cfg/voc.data cfg/yolov3-voc.cfg darknet53.conv.74

训练过程中在backup文件中生成权重文件,本人用时差不多一天,但是模型训练不好,IOU为nan,正常情况应该是一个接近1的值,git的issue里有类似情况,需要进行优化。

GitHub - pjreddie/darknet: Convolutional Neural Networks

模型训练完,可以进行检测,单图检测,选出一张图像放入darknet文件夹下,然后运行:

./darknet detector test cfg/voc.data cfg/yolov3-voc.cfg backup/yolov3-voc_10000.weights 000275.jpg

 然后使用测试集的全部图像进行预测,进而得出准确率。

./darknet detector valid cfg/voc.data cfg/yolov3-voc.cfg backup/yolov3-voc_10000.weights

简单记录了一下安装配置测试过程,后面还有很多工作要做,如:yolov3测试自己数据集,优化测试VOC数据集,测试coco数据集等等~后面继续更新记录

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/124909.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

VSCode无密码连接远程服务器,并能debug python代码

1.官网下载VScode 官网 2.打开VScode&#xff0c;在扩展中搜索下载远程连接插件Remote-SSH 下载完毕会在侧边栏产生“远程资源管理器”图标①&#xff0c;打开远程资源管理器&#xff0c;点击右上角设置进入配置界面&#xff0c;并按照②添加远程服务器账号&#xff0c;输入…

Charles -证书过期失效处理方法

当出现环境配置正常但却无法抓包的时候&#xff0c;可能是因为证书失效了&#xff0c;这种情况移除旧证书&#xff0c;安装新的证书即可。 一、判断是否证书过期 iOS手机&#xff1a; 进入&#xff1a;设置 > 通用 > VPN与设备管理 > Charles Proxy CA... > 更多…

45. 含并行连结的网络(GoogLeNet)代码实现

1. Inception块 import torch from torch import nn from torch.nn import functional as F from d2l import torch as d2lclass Inception(nn.Module):# c1--c4是每条路径的输出通道数,c2,c3,c4是一个tuple元组def __init__(self, in_channels, c1, c2, c3, c4, **kwargs):su…

C# .Net MVC框架实现最简单的登陆

提示&#xff1a;文章写完后&#xff0c;目录可以自动生成&#xff0c;如何生成可参考右边的帮助文档 文章目录前言C#.net实现最简单登陆一、C#,.net是什么&#xff0c;相比较于C,java二、C# .net创建一个MVC框架工程1.步骤实现结果前言 C#.net实现最简单登陆 提示&#xff1…

linux-----基本操作指令(2)

将一个文件按照升序排序:注意这里面的S是大写 按照文件大小按照降序排列:ll -hS 按照文件大小按照升序排列:ll -hrS 一:cp(copy)表示复制&#xff0c;类似于windows系统上面的复制文件到指定文件夹的操作时类似的&#xff0c;拿鼠标一拖到指定路径 1)同时也就是说这个文件最终在…

Python实现的通用的二进制数据分析工具,分析任意格式的二进制数据,还能同时查看协议文档

这是一个通用的二进制数据分析工具。 完整程序代码下载地址&#xff1a;Python实现的通用的二进制数据分析工具 它能做什么 分析任意格式的二进制数据&#xff0c;还能同时查看协议文档逐字节、逐位分析手动、自动分析对分析结果建透视图&#xff0c;发现规律&#xff0c;学习…

IO流的节点流和处理流(缓冲流)and ZIP流使用

流的名称 字节流和字符流的区别 每次读写的字节数不同&#xff1b; 字符流是块读写&#xff0c;字节流是字节读写&#xff1b; 字符流带有缓存&#xff0c;字节流没有 java流在处理上分为字符流和字节流。字符流处理的单元为2个字节的Unicode字符&#xff0c;分别操作字符、…

scipy.interpolate插值方法介绍

文章目录scipy.interpolate插值方法1 一维插值2 multivariate data3 Multivariate data interpolation on a regular grid4 Rbf 插值方法scipy.interpolate插值方法 1 一维插值 from scipy.interpolate import interp1d 1维插值算法 from scipy.interpolate import interp1d…

u盘损坏后如何恢复数据?

u盘经常被用作移动数据存储盘。可以通过USB接口将u盘插入电脑&#xff0c;将需要复制的数据发送到u盘;然后拔掉u盘&#xff0c;插入另一台电脑的USB口打开&#xff0c;就可以把数据复制到另一台电脑上了。实现数据传输。但有使用就意味着有意外&#xff0c;当我们使用中出现了u…

Flowable教程

文章目录一、Flowable介绍1.简介2.Activiti、Flowable、Camunda二、Flowable实战&#xff08;集成Flowable Modeler&#xff09;三、流程的创建和使用1.BPMN基本概念介绍2.业务模型流程创建3.表单创建及使用4.流程的使用5.核心表介绍四、常见报错解决一、Flowable介绍 1.简介 …

3.JMeter基本组成部分

文章目录2.3 JMeter基本组成部分2.3 .1线程组2.3.1.1 添加线程组2.3.1.2 线程组的特点2.3.1.3 线程组分类2.3.1.4 线程组属性2.3.2 取样器 HTTP请求2.3.3 查看结果树2.3 JMeter基本组成部分 2.3 .1线程组 线程组是控制JMeter将用于执行测试的线程组数&#xff0c;也可以把一个…

代理模式

“接口隔离”模式 在组件构建过程中&#xff0c;某些接口之间直接的依赖常常会带来很多问题、甚至根本无法实现。采用添加一层间接&#xff08;稳定&#xff09;接口&#xff0c;来隔离本来互相紧密关联的接口是一种常见的解决方案。 典型模式 ①门面模式 ②代理模式 ③适配…

[oeasy]python0033_回车_carriage_return_figlet_字体变大

回到开头 回忆上次内容 进程前后台切换 ctrl z 把当前进程切换到后台并暂停jobs 查看所有作业 用 fg 可以把后台进程再切回前台 fg %1 可以把指定的任务切回前台用 bg 可以让进程在后台运行 进程查询 ps -elf 查看所有进程信息ps -lf 查看本终端相关进程信息kill -9 PID 给进…

部署项目到Nginx

目录 1、将vue脚手架项目打包 2、将服务端项目打为jar包后上传到linux 3、 使用nginx解决跨域问题 5、 proxy_pass配置问题 1、将vue脚手架项目打包 运行:npm run build命令将vue cli项目打包。 路径在终端会显示 在虚拟机上将此文件上传入nginx中 然后打开nginx,显示页面…

linux下以rpm包安装mysql

目录 下载 上传解压 安装 编辑my.cnf配置文件 启动数据库 修改初始密码,授权远程登录 常见问题 下载 访问以下地址下载对应操作系统下所需的版本 https://downloads.mysql.com/archives/community/ 上传解压 使用ftp或者rz命令上传压缩包到服务器并执行 tar -xvf 压…

JavaScript手写响应式原理(详解)

响应式原理 首先我们有一个对象 const obj {name: zlk,age: 18}这个对象可能在别处被用到 比如是这样的 function foo() {const newValue obj.nameconsole.log(hello world);console.log(obj.name);}我们来改变obj对象中的name的值 obj.name zlk这时候foo()应该被重新执…

Android设计模式详解之代理模式

前言 代理模式也称为委托模式&#xff0c;是一种结构型设计模式&#xff1b; 定义&#xff1a;为其他对象提供一种代理以控制对这个对象的访问&#xff1b; 使用场景&#xff1a;当无法或不想直接访问某个对象或访问某个对象存在困难时&#xff0c;可以通过一个代理对象来间…

css实现圆环、渐变色圆环的多种方式

css实现圆环、渐变色圆环的多种方式一、实现圆环方法具体如下&#xff1a;1. 两个div标签的叠加2.使用伪元素&#xff0c;before&#xff0f;after3. 使用border4. 使用border-shadow5. 使用radial-gradient二、实现渐变色圆环方法具体如下&#xff1a;1.background:linear-gra…

详细记录拉链表的实现过程

面试中被问到了&#xff0c;想了会儿思路混乱没答好&#xff0c;还是理解的不够深刻&#xff0c;重新好好理解记录一下~ 拉链表的用途&#xff0c;主要是用来在数仓中记录业务库数据的全部历史信息和当前最新信息&#xff0c;也就是用来实现对渐变维的记录。数仓中对渐变维的记…