IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> 开发工具 -> [jetson nano]在ubuntu18.04下,c++调用链接库实现yolov3 -> 正文阅读

[开发工具][jetson nano]在ubuntu18.04下,c++调用链接库实现yolov3

目录

0. 前言

1. 下载安装opencv 3.4.0

1.1 配置相应的以来库

1.2 下载opencv 3.4.0(源码)

1.3 编译(时间较长)

1.4 安装

1.5 配置opencv路径

1.6 测试opencv3.4.0

2. c++使用darknet链接库调用yolov3

2.1 准备工作

下图中,可以看到存在darknet、libdarknet.so两个文件(网上有很多darknet资源)?2.2 在终端创建一个新文件夹yolov3_test

2.3 丰富yolov3_test

2.4 编写源码

2.5 编译

2.6 测试效果

3. 后续目标


0. 前言

在正文之前,交代背景。不同与PC端的是,这里采用的是Jetson nano上位机(我感觉与树霉派差不多)。Jeston nano上搭载的是ubuntuu18.04,至于系统问题请自行搜索。

系统:Ubuntu 18.04

软件:opencv 3.4.0、CMake

外设(必须):5V2A直流电源线、摄像头、屏幕、键盘、鼠标

1. 下载安装opencv 3.4.0

先附上两篇参考博客,我按照这两篇博客操作下来是没问题的:第一篇第二篇

以下内容几乎是复制粘贴:

1.1 配置相应的以来库

sudo apt-get install build-essential
# 必须的,gcc编译环境
?
sudo apt-get install cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev
# 必须的,包括cmake等工具
?
sudo apt-get install python-dev python-numpy libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff-dev libjasper-dev libdc1394-22-dev
# 可选的,看个人需要,总共5M左右

1.2 下载opencv 3.4.0(源码)

官网地址:https://opencv.org/releases.html

1.3 编译(时间较长)

cd opencv-3.4.0
mkdir build
# 创建编译文件
cd build
cmake -D CMAKE_BUILD_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local ..
# 生成后的Makefile在build文件夹中,而需要的CMakeLists.txt在 ".."中,即上一级文件夹中(opencv-3.4.0)

1.4 安装

# 4个线程编译,查看cpu核数量
# grep "model name" /proc/cpuinfo | wc -l?
make -j4
?
# 安装
sudo make install

1.5 配置opencv路径

#终端中输入

sudo vim /etc/ld.so.conf.d/opencv.conf

在打开的文件内添加:

/usr/local/lib

#终端中输入

sudo ldconfig

sudo vim /etc/bash.bashrc

在打开的文件夹末尾添加:

PKG_CONFIG_PATH=$PKG_CONFIG_PATH:/usr/local/lib/pkgconfig
export PKG_CONFIG_PATH

#终端中输入

source /etc/bash.bashrc

1.6 测试opencv3.4.0

#上一个终端不用关闭,继续输入

cd ../samples/cpp/example_cmake
cmake .
make
./opencv_example

2. c++使用darknet链接库调用yolov3

2.1 准备工作

下图中,可以看到存在darknet、libdarknet.so两个文件(网上有很多darknet资源)
2.2 在终端创建一个新文件夹yolov3_test

?创建完成后,在文件夹内创建CMakeLists.txt文件,并将以下内容粘贴到文件里。

cmake_minimum_required(VERSION 2.8) # 最低版本需求
project(yolov3) #项目名
#opencv
add_definitions(-std=c++11) 
ADD_DEFINITIONS(-DOPENCV)
ADD_DEFINITIONS(-DGPU)
#########   opencv   #########
set(OpenCV_DIR "/home/zzp/opencv-3.4.0")
find_package( OpenCV REQUIRED )
include_directories( ${OpenCV_INCLUDE_DIRS} )
#########         darknet   #########
include_directories(/home/zzp/darknet-master/include)
find_library(darknet libdarknet.so /home/zzp/darknet-master)
add_executable(${PROJECT_NAME} "/home/zzp/yolov3/yolov3_test.cpp" )
target_link_libraries(${PROJECT_NAME} ${OpenCV_LIBS}  ${darknet})

接下来,在yolov3_test文件夹内新建cfg、img、data、include、weights文件夹,结果如下图所示。?

2.3 丰富yolov3_test

回到darknet-master文件夹,将coco.names、yolov-tiny.cfg放入cfg文件,将dog.jpg放入img,将yolo_v2_class.hpp放入Include,将yolov3-tiny.weights放入weights。

2.4 编写源码

#include <opencv2/opencv.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/highgui.hpp>
#include <vector>
#include <iostream>
#include <fstream>
#include "/home/zzp/yolov3/include/yolo_v2_class.hpp"//引用yolo动态链接库中的头文件

using namespace std;
using namespace cv;

#pragma comment(lib, "yolo_cpp_dll.lib")//引用yolo动态链接库

//画出检测框和相关信息
void DrawBoxes(Mat &frame, vector<string> classes, int classId, float conf, int left, int top, int right, int bottom)
{
	//画检测框
	rectangle(frame, Point(left, top), Point(right, bottom), Scalar(255, 178, 50), 3);
	//该检测框对应的类别和置信度
	string label = format("%.2f", conf);
	if (!classes.empty())
	{
		CV_Assert(classId < (int)classes.size());
		label = classes[classId] + ":" + label;
	}
	//将标签显示在检测框顶部
	int baseLine;
	Size labelSize = getTextSize(label, FONT_HERSHEY_SIMPLEX, 0.5, 1, &baseLine);
	top = max(top, labelSize.height);
	rectangle(frame, Point(left, top - round(1.5*labelSize.height)), Point(left + round(1.5*labelSize.width), top + baseLine), Scalar(255, 255, 255), FILLED);
	putText(frame, label, Point(left, top), FONT_HERSHEY_SIMPLEX, 0.75, Scalar(0, 0, 0), 1);
}

//画出检测结果
void Drawer(Mat &frame, vector<bbox_t> outs, vector<string> classes)
{
	//获取所有最佳检测框信息
	for (int i = 0; i < outs.size(); i++)
	{
		DrawBoxes(frame, classes, outs[i].obj_id, outs[i].prob, outs[i].x, outs[i].y,
			outs[i].x + outs[i].w, outs[i].y + outs[i].h);
	}
}

int main(void)
{
	string classesFile = "/home/zzp/yolov3/cfg/coco.names";
	string modelConfig = "/home/zzp/yolov3/cfg/yolov3-tiny.cfg";
	string modelWeights = "/home/zzp/yolov3/weights/yolov3-tiny.weights";

	//加载类别名
	vector<string> classes;
	ifstream ifs(classesFile.c_str());
	string line;
	while (getline(ifs, line)) classes.push_back(line);
	//加载网络模型,0是指定第一块GPU
	Detector detector(modelConfig, modelWeights, 0);

	string mode = "video";
	//图像
	if (mode == "image")
	{
		Mat frame = imread("/home/zzp/yolov3/img/test.jpeg");
		std::vector<bbox_t> result_vec = detector.detect(frame);
		Drawer(frame, result_vec, classes);
		cv::namedWindow("CV_WINDOW", CV_WINDOW_NORMAL);
		cv::imshow("CV_IMAGE", frame);
		cv::imwrite("/home/zzp/yolov3/img/result.jpg",frame);
		cv::waitKey(0);
	}
	//视频
	else if (mode == "video")
	{
		VideoCapture cap(0);
		Size size(640, 480);
		VideoWriter writer("/home/zzp/yolov3/data/result.avi", cv::VideoWriter::fourcc('M', 'J', 'P', 'G'), 25, size);
		while (1)
		{
			Mat frame;
			cap >> frame;
			if (frame.empty()) 
				break;

			std::vector<bbox_t> result_vec = detector.detect(frame);
			//画图
			Drawer(frame, result_vec, classes);
			cv::imshow("YOLO V3 Camera", frame);
			for(int i = 0; i < result_vec.size(); i++)
			{    
                //检测到人就告知用户此人所在区域
				if(result_vec[i].obj_id == 0)
				{
					cout << "-----------------------------------------------------------------------" << endl;
					cout << "检测到一个人,他(她)位于视频中这个位置: " << endl;
					cout << "起始点(左上):(" << result_vec[i].x << "," << result_vec[i].y << ")"
						<< "\t宽与高:(" << result_vec[i].w << "," << result_vec[i].h << ")." << endl; 
					cout << "-----------------------------------------------------------------------" << endl;
				}
			}
            //ESC退出程序
			if(waitKey(3) == 27) break;
			writer << frame;
		}
		cv::destroyAllWindows();
		cap.release();
	}
    return 0;
}

2.5 编译

此时yolov3_test文件夹有如下文件:

#在终端中输入

cd yolov3_test

cmake .

make

结果如下图所示:

2.6 测试效果

?#测试图片(将源码string mode修改为 image)

cd yolov3_test

./yolov3

#测试视频(自己测试吧,我这里是没毛病的!)

3. 后续目标

探索Jetson nano串口功能,实现检测到目标后实时反馈数据。

  开发工具 最新文章
Postman接口测试之Mock快速入门
ASCII码空格替换查表_最全ASCII码对照表0-2
如何使用 ssh 建立 socks 代理
Typora配合PicGo阿里云图床配置
SoapUI、Jmeter、Postman三种接口测试工具的
github用相对路径显示图片_GitHub 中 readm
Windows编译g2o及其g2o viewer
解决jupyter notebook无法连接/ jupyter连接
Git恢复到之前版本
VScode常用快捷键
上一篇文章      下一篇文章      查看所有文章
加:2022-02-09 20:54:32  更:2022-02-09 20:55:13 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 -2024/11/15 10:12:00-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码