C++ 代码示例

本章会详细介绍DaoAI World SDK中包含的C++代码示例。

您也可以查看我们的 GitHub repo，其中包含 C++、C# 和 Python 的示例项目，方便用户快速上手和参考。

引入库 

在C++示例中，我们使用了以下几个头文件，其中 dlsdk/model.h 是用于引入DaoAI World SDK的库。

#include <dlsdk/model.h>
#include <dlsdk/prediction.h>
#include <string>
#include <fstream>

读取图片 

DaoAI World SDK 的模型预测函数需要将图片表示为一维数组（1D array）。以下是从文件中读取图片的函数：

首先定义文件路径，然后调用函数读取。可以调用daoai_image()方法读取图片。

// get root path to the model and image
std::string root = "../"; // change to your own path
std::string image_path = root + "image.png";

// load image
DaoAI::DeepLearning::Image daoai_image(image_path);

加载深度学习模型 

首先需要加载模型。DaoAI World 输出的深度学习模型通常是 dwm 格式。我们需要创建一个 DaoAI::DeepLearning::Vision::InstanceSegmentatio 对象，然后使用constructor方法来读取 DaoAI World 输出的深度学习模型 dwm 文件。

std::string root = "../"; // change to your own path
        std::string model_path = root + "model.dwm";

// init model
        DaoAI::DeepLearning::Vision::InstanceSegmentation model(model_path);

注意，这里每一个检测任务都有对应的对象：

//实例分割
DaoAI::DeepLearning::Vision::InstanceSegmentation model(model_path);

//关键点检测
DaoAI::DeepLearning::Vision::KeypointDetection model(model_path);

//图像分类
DaoAI::DeepLearning::Vision::Classification model(model_path);

//目标检测
DaoAI::DeepLearning::Vision::ObjectDetection model(model_path);

//非监督缺陷检测 需要用 DaoAI 非监督 SDK

//监督缺陷检测
DaoAI::DeepLearning::Vision::SupervisedDefectSegmentation model(model_path);

//OCR
DaoAI::DeepLearning::Vision::OCR model(model_path);

//定位模型 (只在工业版支持)
DaoAI::DeepLearning::Vision::Positioning model(model_path);

//漏错装检测 (只在工业版支持)
DaoAI::DeepLearning::Vision::PresenceChecking model(model_path);

如果尝试加载非对应的模型对象，那么会报错，报错信息中会提示您应该用的模型类型。

使用深度学习模型进行预测 

备注

在第一次运行时，需要加载模型等数据到内存，这通常需要更久的时间。在程序的后续运行中，也就是从第二次图片加载，推理时，运行时间就会稳定在更快速的时间。

// get inference
DaoAI::DeepLearning::Vision::InstanceSegmentationResult prediction = model.inference(daoai_image);

//std::vector<DaoAI::DeepLearning::Polygon> polygons = prediction.masks[1].toPolygons();
std::string json_string = prediction.toJSONString(); // 标准输出的Json
std::string annotation_json_string = prediction.toAnnotationJSONString(); //按照数据的标注的格式输出Json
// write to json file
std::ofstream fout(root + "daoai_1.json");
fout << json_string << "\n";
fout.close();

注意，这里每一个检测任务返回的结果都有对应的对象：

//实例分割
DaoAI::DeepLearning::Vision::InstanceSegmentationResult prediction = model.inference(daoai_image);

//关键点检测
DaoAI::DeepLearning::Vision::KeypointDetectionResult prediction = model.inference(daoai_image);

//图像分类
DaoAI::DeepLearning::Vision::ClassificationResult prediction = model.inference(daoai_image);

//目标检测
DaoAI::DeepLearning::Vision::ObjectDetectionResult prediction = model.inference(daoai_image);

//非监督缺陷检测 需要用 DaoAI 非监督 SDK

//监督缺陷检测
DaoAI::DeepLearning::Vision::SupervisedDefectSegmentationResult prediction = model.inference(daoai_image);

//OCR
DaoAI::DeepLearning::Vision::OCRResult prediction = model.inference(daoai_image);

//定位模型 (只在工业版支持)
DaoAI::DeepLearning::Vision::PositioningResult prediction = model.inference(daoai_image);

//漏错装检测 (只在工业版支持)
DaoAI::DeepLearning::Vision::PresenceCheckingResult prediction = model.inference(daoai_image);

后处理参数 

模型的预测可以接受后处理参数：

model.setConfidenceThreshold();

置信度阈值，会过滤掉结果中置信度低于设定值的结果范围 0-1

model.setIOUThreshold();

IOU阈值，会过滤掉结果中IOU低于设定值的结果范围 0-1

DaoAI::DeepLearning::Vision::InstanceSegmentationResult prediction = model.inference(daoai_image, {{DaoAI::DeepLearning::PostProcessType::CONFIDENCE_THRESHOLD, 0.4}, {DaoAI::DeepLearning::PostProcessType::IOU_THRESHOLD, 0.5} });

获取预测结果 

Box（边界框）

Box 是模型预测结果中的边界框，可通过以下方式获取：

DaoAI::DeepLearning::Vision::KeypointDetectionResult prediction = model.inference(daoai_image);

prediction.boxes[0].x1();  // 左上角 x 坐标
prediction.boxes[0].y1();  // 左上角 y 坐标
prediction.boxes[0].x2();  // 右下角 x 坐标
prediction.boxes[0].y2();  // 右下角 y 坐标

Mask（掩码）

Mask 是预测结果中目标物体的外轮廓信息，可以通过以下方式提取多边形区域的顶点：

DaoAI::DeepLearning::Vision::KeypointDetectionResult prediction = model.inference(daoai_image);

prediction.masks[0].toPolygons()[0].points[0].x  // 顶点的 X 坐标
prediction.masks[0].toPolygons()[0].points[0].y  // 顶点的 Y 坐标

`masks[0]`：提取第一个目标物体的掩码。
`toPolygons()`：将掩码转换为多边形外轮廓。
`points[0]`：获取多边形的第一个顶点坐标。

通过顺序连接多边形的所有顶点，可以形成完整的轮廓区域，用于目标检测、区域分析等应用场景。

可视化输出 

将预测的掩码图保存为单通道灰度图，可以使用opencv库。

DaoAI::DeepLearning::Image image = prediction.masks[0].toImage();
cv::imwrite("mask.png", cv::Mat(image.rows,image.cols, CV_8UC1, image.getData()));

生成并保存预测结果与原图叠加的可视化图像：

DaoAI::DeepLearning::Image result = DaoAI::DeepLearning::Utils::visualize(daoai_image, prediction);
result.save(root + "daoai_output.png");

输出为Json格式 

以下是实例分割模型预测后调用 toJSONString() 方法所返回的结果示例。

这个结果展示了预测的数量，标签名称，置信度，以及预测框，和多边形掩膜。

{
    "Number of detecions": 1,
    "Detections": [
        {
            "Label": "zheng",
            "Confidence": 0.9523001313209534,
            "Box": [
                955.1925659179688, 316.0162048339844, 1064.072021484375,
                426.4408264160156
            ],
            "Mask": [
                [990.0, 316.0],
                [988.0, 318.0],
                [987.0, 318.0],
                [985.0, 320.0],
                [982.0, 320.0],
                [980.0, 322.0],
                [979.0, 322.0],
                [974.0, 327.0],
                [972.0, 327.0],
                [972.0, 328.0],
                [1040.0, 316.0]
            ]
        }
    ],
    "ImageHeight": 1200,
    "ImageWidth": 1920
}