OpenCV4.8 Java SDK實(shí)現(xiàn)YOLOv5模型部署

OpenCV Java開發(fā)環(huán)境配置

Java 開發(fā)的IDE，推薦選擇IDEA這個(gè)，真的比較好用，我下載的是社區(qū)版，免費(fèi)可用。

安裝好以后直接點(diǎn)擊打開，選擇 New Project

添加本地的OpenCV Java SDK的 Jar文件依賴包 從Project Structure到 Modules 到 Module SDK 添加Jars or Dependence

OpenCV Java 代碼測(cè)試

OpenCV Java配置好以后，測(cè)試代碼首先需要加載DLL，然后依賴Java SDK，可以實(shí)現(xiàn)圖像加載與顯示。測(cè)試代碼如下：

package org.example;


import org.opencv.core.*;
import org.opencv.highgui.HighGui;
import org.opencv.imgcodecs.Imgcodecs;




//TIP To Run code, press  or
// click the  icon in the gutter.
public class YOLOv5DNNDetector {
    public static void main(String[] args) {
        System.load("D:/opencv-4.8.0/opencv/build/java/x64/opencv_java480.dll");
        System.out.println("start to read image...");
        Mat inputImage = Imgcodecs.imread("D:/images/lena.jpg");
        HighGui.imshow("Java OpenCV開發(fā)測(cè)試", inputImage);
        HighGui.waitKey(0);
        HighGui.destroyAllWindows();
        System.exit(0);
    }
}

運(yùn)行結(jié)果如下：

YOLOv5模型部署

基于YOLOv5 7.0版本，完成了對(duì)象檢測(cè)

publicstaticvoidmain(String[]args){

System.load("D:/opencv-4.8.0/opencv/build/java/x64/opencv_java480.dll");
System.out.println("starttoreadimage...");
Stringfile_path="D:/python/yolov5-7.0/classes.txt";
try{
MatinputImage=Imgcodecs.imread("D:/bird_test/city-walk.jpg");
java.util.ArrayListboxes=newArrayList<>();
FileInputStreamfis=newFileInputStream(file_path);
BufferedReaderbr=newBufferedReader(newInputStreamReader(fis));
YOLOv5DNNDetectordetector=newYOLOv5DNNDetector("D:/python/yolov5-7.0/yolov5s.onnx",0.5f,0.5f,640,640);
Stringline;
ArrayListmap_labels=newArrayList<>();
while((line=br.readLine())!=null){
map_labels.add(line);
}
detector.addAllLabels(map_labels);
detector.infer_image(inputImage,boxes);
HighGui.imshow("OpenCV4.8forJavaYOLOv5對(duì)象檢測(cè)演示",inputImage);
HighGui.waitKey(0);
HighGui.destroyAllWindows();
}catch(Exceptione){
e.printStackTrace();
}
System.exit(0);
}