快速入门

本章介绍如何应用Mech-DLK SDK实现Mech-DLK训练的缺陷分割模型的推理。

准备工作

安装 Mech-DLK SDK。
下载并安装Sentinel LDK加密驱动，并确保在工控机的设置应用 应用和功能中存在已安装的加密驱动。

如已安装Mech-DLK，则不需要重复安装加密驱动，只需确认加密驱动已存在即可。
获得并管理软件许可。

推理流程

函数说明

下面以Mech-DLK导出的缺陷分割模型为例，介绍应用Mech-DLK SDK实现模型推理时使用的函数。

创建输入图像

调用以下函数创建输入图像。

MMindImage image = new MMindImage();
image.CreateFromPath("path/to/image.png");
List<MMindImage> images = new List<MMindImage> { image };

mmind::dl::MMindImage image;
image.createFromPath(“path/to/image.png”);
std::vector<mmind::dl::MMindImage> images = {image};

MMindImage input;
createImage("path/to/image.png", &input);

创建推理引擎

调用以下函数创建创建推理引擎。

InferEngine inferEngine = new InferEngine();
inferEngine.Create("path/to/xxx.dlkpack", BackendType.GpuDefault, 0);

当部署电脑上配置有英伟达的独立显卡时，你可以将推理后端（即第二个参数）设置为GpuDefault或GpuOptimization。
- 当此参数设置为GpuOptimization时，需要等待1~5分钟的模型优化时间，并且浮点精度FP16只在该模式下生效。
当部署电脑上未配置英伟达的独立显卡时，推理后端只能设置为CPU。
此函数中，第三个参数0为英伟达显卡ID，如果只有一张显卡，填写0即可。当推理后端设置为CPU时，此参数无效。

mmind::dl::MMindInferEngine engine;
engine.create(kPackPath);
// engine.setInferDeviceType(mmind::dl::InferDeviceType::GpuDefault);
// engine.setBatchSize(1);
// engine.setFloatPrecision(mmind::dl::FloatPrecisionType::FP32);
// engine.setDeviceId(0);
engine.load();

C++接口中模型参数可自行设置：

如果不调用setxxx函数，默认情况下BatchSize为1；FloatPrecision为FP32；DeviceId为0。
如果当前设备具备英伟达的独立显卡，那么InferDeviceType为GpuDefault；否则，InferDeviceType为CPU。
如需修改推理引擎参数，setxxx函数必须放在load()函数之前。
当InferDeviceType设置为GpuOptimization时，需要等待1~5分钟的模型优化时间，并且浮点精度FP16只在该模式下生效。

Engine engine;
createPackInferEngine(&engine, "path/to/xxx.dlkpack", GpuDefault, 0);

当部署电脑上配置有英伟达的独立显卡时，你可以将推理后端（即第三个参数）设置为GpuDefault或GpuOptimization。
- 当此参数设置为GpuOptimization时，需要等待1~5分钟的模型优化时间。
当部署电脑上未配置英伟达的独立显卡时，推理后端只能设置为CPU。
此函数中，第四个参数0为英伟达显卡ID，如果只有一张显卡，填写0即可。当推理后端设置为CPU时，此参数无效。

深度学习引擎推理

调用以下函数进行深度学习引擎推理。

inferEngine.Infer(images);

engine.infer(images);

infer(&engine, &input, 1);

此函数中，参数1表示推理的图片数量，必须与input中的图片数量相同。

获取缺陷分割结果

调用以下函数得到缺陷分割模型的结果。

List<Result> results;
inferEngine.GetResults(out results);

std::vector<mmind::dl::MMindResult> results;
engine.getResults(results);

DefectAndEdgeResult* defectAndEdgeResult = NULL;
unsigned int resultNum = 0;
getDefectSegmentataionResult(&engine, 0, &defectAndEdgeResult, &resultNum);

此函数中，第二个参数0表示深度学习模型推理包中的模型索引。

如果是单模型推理包，该参数只能设置为0；
如果为级联模型推理包，该参数需要根据模型推理包中算法的顺序进行设置。

结果可视化

调用以下函数可视化模型推理结果。

inferEngine.ResultVisualization(images);
image.Show("result");

engine.resultVisualization(images);
image.show("Result");

resultVisualization(&engine, &input, 1);
showImage(&input, "result");