Linux環境下使用 C++ 與 OpenCV 實現 ONNX 分類模型推理

實驗環境:Ubuntu 20.0
推理模型:ONNX分類模型

1. 安裝依賴項

首先是需要安裝依賴庫,如g++cmake等,如果已經安裝的話可以忽略

sudo apt install -y g++
sudo apt install -y cmake
sudo apt install -y make
sudo apt install -y wget unzip

安裝一些opencv的依賴項(可不安裝)

sudo apt-get install build-essential libgtk2.0-dev libgtk-3-dev libavcodec-dev libavformat-dev libjpeg-dev libswscale-dev libtiff5-dev

安裝一些可選庫(可不安裝)

# streamer支持
sudo apt install libgstreamer-plugins-base1.0-dev libgstreamer1.0-dev
# 可選的依賴
sudo apt install libpng-dev libopenexr-dev libtiff-dev libwebp-dev

在這里插入圖片描述

2. 下載并安裝OpenCV

可以在官網下載相應版本的OpenCV,主要有SourceGitHub兩種方式下載。

2.1 Source下載

Source:https://opencv.org/releases/

點擊Source進行下載:
在這里插入圖片描述

2.2 GitHub下載

這里推薦安裝4.9.0版本。

# 安裝4.9.0版本
wget -O opencv.zip https://github.com/opencv/opencv/archive/4.9.0.zip
# 安裝最新版本
wget -O opencv.zip https://github.com/opencv/opencv/archive/master.zip

下載完成后解壓,并將文件夾名字改為opencv

在這里插入圖片描述

3. Cmake配置和編譯OpenCV

進入到下載好的opencv目錄中,新建并進入目錄build

cd opencv
mkdir build
cd build

使用cmake配置opencv

cmake -D CMAKE_BUILD_TYPE=Release -D OPENCV_GENERATE_PKGCONFIG=YES ..

在這里插入圖片描述

說明:
-D OPENCV_GENERATE_PKGCONFIG=YES OpenCV4以上默認不使用pkg-config,該編譯選項開啟生成opencv4.pc文件,支持pkg-config功能

隨后使用make進行編譯

make -j4

在這里插入圖片描述

說明:
-j4中的4指同時使用4個進程,可以根據電腦的進程數調整此值

make進行安裝(可以不用sudo)

sudo make install

在這里插入圖片描述

默認安裝路徑為:
/usr/local/bin - executable files
/usr/local/lib - libraries(.so)
/usr/local/cmake/opencv4 - cmake package
/usr/local/include/opencv4 - headers
/usr/local/share/opencv4 - other files (e.g. trained cascades in XML format)

4. 環境配置

4.1 配置pkg-config環境

opencv4.pc文件的默認路徑:/usr/local/lib/pkgconfig/opencv4.pc
若此目錄下沒有,可以使用以下命令搜索:

sudo find / -iname opencv4.pc

可以看到在該目錄下面:
在這里插入圖片描述
將路徑加入到PKG_CONFIG_PATH(用vim打開):

sudo vim /etc/profile.d/pkgconfig.sh

在文件后面加入下面一行:

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

保存并退出后激活:
vim保存退出的方式:按ESC 輸入:wq 然后回車

# 激活
source /etc/profile

用以下命令驗證是否成功:

pkg-config --libs opencv4

輸出結果如下則代表成功

在這里插入圖片描述

4.2 配置動態庫環境

打開文件(可能為空文件):

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

在該文件末尾加上OpenCVlib路徑,保存退出:

/usr/local/lib

使配置的路徑生效:

sudo ldconfig

5. 測試OpenCV(CPU)

cd/opencv/samples/cpp/example_cmake目錄下,然后打開opencv_example.cpp文件,替換為下面的代碼,這個代碼執行的便是 加載分類ONNX模型進行推理。

這里博主使用的是OpenCV中的dnn.hpp來加載ONNX模型,當然我們也可以選擇使用ONNX Runtime

#include <opencv2/opencv.hpp>
#include <opencv2/dnn.hpp>
#include <iostream>
#include <vector>
#include <cmath>// Softmax 函數
void softmax(const float* data, float* output, int size) {float max_val = -INFINITY;for (int i = 0; i < size; ++i) {max_val = std::max(max_val, data[i]);}float sum_exp = 0.0f;for (int i = 0; i < size; ++i) {sum_exp += std::exp(data[i] - max_val);}for (int i = 0; i < size; ++i) {output[i] = std::exp(data[i] - max_val) / sum_exp;}
}int main() {// 加載模型std::string modelPath = "juan_cls.onnx";cv::dnn::Net net = cv::dnn::readNetFromONNX(modelPath);if (net.empty()) {std::cerr << "無法加載模型,請檢查路徑是否正確" << std::endl;return -1;}// 圖像輸入cv::Mat img = cv::imread("img.jpg");if (img.empty()) {std::cerr << "無法加載圖像 img.jpg" << std::endl;return -1;}// 預處理int inputWidth = 640;int inputHeight = 640;cv::Mat blob = cv::dnn::blobFromImage(img, 1.0, cv::Size(inputWidth, inputHeight),cv::Scalar(), true, false);net.setInput(blob);// 推理cv::Mat out = net.forward();std::cout << "Inference Shape: " << out.size << std::endl;// 假設 out 是 [1 x C x H x W]int batchSize = out.size[0];   // 1int channels = out.size[1];    // 2int height = out.size[2];      // 640int width = out.size[3];       // 640std::vector<float> avgScores(channels, 0.0f);for (int c = 0; c < channels; ++c) {double sum = 0.0;float* ptr = out.ptr<float>(0, c);  // batch=0, channel=cfor (int h = 0; h < height; ++h) {for (int w = 0; w < width; ++w) {sum += ptr[h * width + w];}}avgScores[c] = static_cast<float>(sum / (height * width));}// Softmax 歸一化float probs[2];softmax(avgScores.data(), probs, 2);// 輸出結果std::cout.precision(4);std::cout << std::fixed;std::cout << "\n【Result】" << std::endl;std::cout << "Cls 0 Score: " << probs[0] << std::endl;std::cout << "Cls 1 Score: " << probs[1] << std::endl;std::cout << "Cls: " << (probs[0] > probs[1] ? 0 : 1) << std::endl;return 0;
}

隨后,進行編譯執行:

cmake .
make
./opencv_example

cmake .后,會生成下面這些文件,根據 CMakeLists.txt 生成編譯規則(如 Makefile)
在這里插入圖片描述

make根據 Makefile 編譯出可執行文件

在這里插入圖片描述

./opencv_example運行編譯好的程序,執行結果如下:

在這里插入圖片描述

至此,我們便已經完成了使用C++OpenCV加載ONNX模型進行推理,接下來,我們將調用GPU來進行加速推理,敬請期待。

本文來自互聯網用戶投稿,該文觀點僅代表作者本人,不代表本站立場。本站僅提供信息存儲空間服務,不擁有所有權,不承擔相關法律責任。
如若轉載,請注明出處:http://www.pswp.cn/news/912818.shtml
繁體地址,請注明出處:http://hk.pswp.cn/news/912818.shtml
英文地址,請注明出處:http://en.pswp.cn/news/912818.shtml

如若內容造成侵權/違法違規/事實不符,請聯系多彩編程網進行投訴反饋email:809451989@qq.com,一經查實,立即刪除!

相關文章

AJAX 安裝使用教程

一、AJAX 簡介 AJAX&#xff08;Asynchronous JavaScript and XML&#xff09;是一種在無需重新加載整個網頁的情況下&#xff0c;能夠與服務器交換數據并更新部分網頁內容的技術。它不是一種新語言&#xff0c;而是使用現有的標準組合&#xff1a;JavaScript XMLHttpRequest…

【牛客算法】牛客網編程題解:小紅拼圖

一、題目介紹 1.1. 題目鏈接 &#xff1a;小紅拼圖 https://www.nowcoder.com/questionTerminal/08b54686f0d14bd784d9d148c68a268a 1.2 題目介紹 小紅正在玩一個拼圖游戲&#xff0c;她有一些完全相同的拼圖組件&#xff1a; 小紅準備用這些組件來拼成一些圖案。這些組件可…

買賣股票的最佳時機--js 算法

一、買賣股票的最佳時機 給定一個數組 prices ,它的第 i 個元素 prices[i] 表示一支給定股票第 i 天的價格。你只能選擇 某一天 買入這只股票,并選擇在 未來的某一個不同的日子 賣出該股票。設計一個算法來計算你所能獲取的最大利潤。返回你可以從這筆交易中獲取的最大利潤。…

C#基礎(WndProc)

WndProc 是操作系統與你的程序“對話”的通道??。當用戶點擊鼠標、按下鍵盤&#xff0c;或系統事件&#xff08;如窗口移動&#xff09;發生時&#xff0c;Windows 會將這些事件打包成“消息”&#xff0c;發送給你的窗口&#xff0c;而 WndProc 就是接收和處理這些消息的函數…

記錄一個 Linux中腳本無法執行的問題

問題描述&#xff1a; 在本地的window系統傳的云服務器上一個.sh結尾的安裝Java環境的腳本 上傳到云服務器后&#xff0c;使用命令賦予執行權限 chmod x 文件名然后看一下這個腳本變綠了就可以了 然后開始嘗試執行 ./腳本名 然后就報錯了 然后開始排查問題 1.檢查并修復 She…

Iceberg在圖靈落地應用

導讀 百度MEG上一代大數據產品存在平臺分散、易用性差等問題&#xff0c;導致開發效率低下、學習成本高&#xff0c;業務需求響應遲緩。為了解決這些問題&#xff0c;百度MEG內部開發了圖靈3.0生態系統&#xff0c;包括Turing Data Engine(TDE)計算&存儲引擎、Turing Data…

FPGA設計的用戶約束

FPGA設計的用戶約束 文章目錄 FPGA設計的用戶約束FPGA設計的用戶約束綜合約束管腳約束位置約束時序約束小總結 FPGA設計的用戶約束 至此&#xff0c;HDL到門級網表的轉化已經完成&#xff0c;對于編譯器來說&#xff0c;下一步的任務就是要將門級網表轉換并映射到具體的FPGA硬…

Spring 生態創新應用:微服務架構設計與前沿技術融合實踐

在數字化轉型的深水區&#xff0c;企業級應用正面臨從 “單體架構” 向 “分布式智能架構” 的根本性躍遷。Spring 生態以其二十年技術沉淀形成的生態壁壘&#xff0c;已成為支撐這場變革的核心基礎設施。從 2002 年 Rod Johnson 發布《Expert One-on-One J2EE Design and Deve…

車牌識別與標注:基于百度OCR與OpenCV的實現(一)

車牌識別與標注&#xff1a;基于百度OCR與OpenCV的實現 在計算機視覺領域&#xff0c;車牌識別是一項極具實用價值的技術&#xff0c;廣泛應用于交通監控、智能停車場管理等領域。本文將介紹如何在macOS系統下&#xff0c;利用百度OCR API進行車牌識別&#xff0c;并結合OpenC…

【系統分析師】2021年真題:論文及解題思路

文章目錄 試題一&#xff1a;論面向對象的信息系統分析方法試題二&#xff1a;論靜態測試方法及其應用試題三&#xff1a;論富互聯網應用的客戶端開發技術試題四&#xff1a;論DevSecOps技術及其應用 試題一&#xff1a;論面向對象的信息系統分析方法 信息系統分析是信息系統生…

OFA-PT:統一多模態預訓練模型的Prompt微調

摘要 Prompt微調已成為模型微調的新范式&#xff0c;并在自然語言預訓練甚至視覺預訓練中取得了成功。參數高效的Prompt微調方法通過優化soft embedding并保持預訓練模型凍結&#xff0c;在計算成本低和幾乎無性能損失方面展現出優勢。在本研究中&#xff0c;我們探索了Prompt…

【硬核數學】2.5 “價值標尺”-損失函數:信息論如何設計深度學習的損失函數《從零構建機器學習、深度學習到LLM的數學認知》

歡迎來到本系列硬核數學之旅的第十篇&#xff0c;也是我們對經典數學領域進行深度學習“升級”的最后一站。我們已經擁有了強大的模型架構&#xff08;基于張量&#xff09;、高效的學習引擎&#xff08;反向傳播&#xff09;和智能的優化策略&#xff08;Adam等&#xff09;。…

雷卯針對靈眸科技EASY EAI nano RV1126 開發板防雷防靜電方案

一、應用場景 1. 人臉檢測 2. 人臉識別 3. 安全帽檢測 4. 人員檢測 5. OCR文字識別 6. 人頭檢測 7. 表情神態識別 8. 人體骨骼點識別 9. 火焰檢測 10. 人臉姿態估計 11. 人手檢測 12. 車輛檢測 13. 二維碼識別 二、 功能概述 1 CPU 四核ARM Cortex-A71.5GHz 2 …

【記錄】Ubuntu|Ubuntu服務器掛載新的硬盤的流程(開機自動掛載)

簡而言之&#xff0c;看這張圖片就好&#xff08;可以存一下&#xff0c;注意掛載點/data可以自定義&#xff0c;掛載硬盤的位置/dev/sdb要改成步驟1中檢查的時候查到的那個位置&#xff0c;不過這個圖的自動掛載漏了UUID&#xff0c;可以通過blkid指令查找&#xff09;&#x…

六、軟件操作手冊

建議在飛書平臺閱讀此文。 我將沿著初來乍到的用戶的瀏覽路徑介紹“諍略參謀”應用。 目錄 一、用戶信息1.1 注冊、登錄、自動登錄、忘記密碼、修改用戶名、修改密碼、退出登錄與個性化設置1.2 認識主界面與任務系統1.3 語義審查、Knowledge Cutoff 審查1.4 重要內容未保存提醒…

電腦鍵盤不能打字了怎么解決 查看恢復方法

電腦鍵盤打不了字&#xff0c;這是我們電腦使用過程中&#xff0c;偶爾會遇到的電腦故障問題。一般來說&#xff0c;電腦鍵盤打不出字&#xff0c;可能是硬件故障、驅動問題或系統設置錯誤等多種原因引起。本文將詳細介紹一些常見的原因和解決方法&#xff0c;幫助用戶恢復正常…

基于STM32的土豆種植自動化灌溉系統設計與實現

?? 項目簡介 隨著農業現代化發展及水資源短缺問題日益突出,傳統土豆種植方式在澆灌效率與用水科學性方面暴露出諸多問題。本文基于STM32F103C8T6微控制器,設計并實現了一種智能化的土豆種植自動灌溉系統,集成多種環境傳感器(溫濕度、土壤濕度、光照)、控制設備(水泵、…

第8篇:Gin錯誤處理——讓你的應用更健壯

作者:GO兔 博客:https://luckxgo.cn 分享大家都看得懂的博客 引言 在Web應用開發中&#xff0c;錯誤處理是保證系統穩定性和用戶體驗的關鍵環節。Gin作為高性能的Go Web框架&#xff0c;提供了靈活的錯誤處理機制&#xff0c;但許多開發者在實際項目中仍會遇到錯誤處理混亂、異…

【PyCharm】Python安裝路徑查找

PyCharm應用筆記 第一章 Python安裝路徑查找 文章目錄 PyCharm應用筆記前言一、電腦設置查找二、資源管理器查找 前言 本文主要介紹幾種Python安裝路徑查找的方法。 一、電腦設置查找 簡述過程&#xff1a;設置》應用》安裝的應用》搜索框輸入Python。 注&#xff1a;電腦使用…

數據結構:遞歸:漢諾塔問題(Tower of Hanoi)

目錄 問題描述 第一性原理分析 代碼實現 第一步&#xff1a;明確函數要干什么 第二步&#xff1a;寫好遞歸的“結束條件” 第三步&#xff1a;寫遞歸步驟 &#x1f333; 遞歸調用樹 &#x1f50d;復雜度分析 時間復雜度&#xff1a;T(n) 2^n - 1 空間復雜度分析 問題描…