快速入门
本章介绍如何应用Mech-DLK SDK实现Mech-DLK训练的缺陷分割模型的推理。
函数说明
下面以Mech-DLK导出的缺陷分割模型为例,介绍应用Mech-DLK SDK实现模型推理时使用的函数。
| 自 Mech-DLK SDK 3.0.0 起,不再支持 C 语言接口。以下示例仅包含 C#、C++ 与 Python。 |
创建输入图像
调用以下函数创建输入图像。
-
C#
-
C++
-
Python
MMindImage image = new MMindImage();
StatusCode status = 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};
import mmind_dl_sdk as mmind
image = mmind.MMindImage()
status = image.create_from_path("path/to/image.png")
images = [image]
创建推理引擎
调用以下函数创建推理引擎。
-
C#
-
C++
-
Python
MMindInferEngine engine = new MMindInferEngine();
engine.Create("path/to/xxx.dlkpack");
StatusCode status = engine.Load();
|
Mech-DLK SDK C# 接口中, |
mmind::dl::MMindInferEngine engine;
engine.create(L"path/to/xxx.dlkpack");
// engine.setInferDeviceType(mmind::dl::InferDeviceType::GPUDefault);
// engine.setDeviceId(0);
// engine.setBatchSize("DefectSegmentation_0", 1);
// engine.setFloatPrecision("DefectSegmentation_0", mmind::dl::PrecisionType::FP32);
engine.load();
|
C++接口中模型参数可自行设置:
|
engine = mmind.MMindInferEngine()
engine.create("path/to/xxx.dlkpack")
status = engine.load()
|
Python 接口中, |
深度学习引擎推理
调用以下函数进行深度学习引擎推理。
-
C#
-
C++
-
Python
engine.Infer(images);
engine.infer(images);
status = engine.infer(images)
获取缺陷分割结果
调用以下函数得到缺陷分割模型的结果。
-
C#
-
C++
-
Python
List<MMindResult> results;
engine.GetModuleResult("DefectSegmentation_0", out results);
std::vector<mmind::dl::MMindResult> results;
engine.getModuleResult("DefectSegmentation_0", results);
status, results = engine.get_module_result("DefectSegmentation_0")
|
|