C#调用WITIAI模型

witiai

C#调用WITIAI模型
（1）WITIAI软件训练好AI模型后，采用C#调用AI模型，得到分割分类图。
（2）halcon、opencv的接口都预留了，满足工业需求；
（3）Demo代码如下：

1.             string ImagePath = this.textBox1.Text.ToString();
   
2.             int sliceImageSize = 512;
   
3.             double overRate = 0.0;
   
4.             double image_threshold = 0.5;
   
5.             int classnum = 4;
   
6.             int GPUindex = 0;
   
7.             int ImageChannel = 1;
   
8.             string savepath1 = System.Windows.Forms.Application.StartupPath + @"/" + "gray_1.png";
   
9.             // 初始化AI模型
   
10.             string AImodelpt = @"gray_model.WITIAI";
    
11.             // method1
    
12.             CallAIDLL.load_module(AImodelpt, GPUindex);
    
13.             CallAIDLL.PredictGrayImage(ImagePath, savepath1, sliceImageSize, overRate, image_threshold, classnum, GPUindex);
    
14. 
    
15.             // method2 -- halcon12进行调试
    
16.             HObject ho_Image = null;
    
17.             HTuple Width = 1200, Height = 1200;
    
18.             HOperatorSet.GenEmptyObj(out ho_Image);
    
19.             ho_Image.Dispose();
    
20.             HOperatorSet.ReadImage(out ho_Image, ImagePath);
    
21.             HOperatorSet.GetImageSize(ho_Image, out Width, out Height);
    
22.             IntPtr p = Marshal.AllocHGlobal(Width.I * Height.I * ImageChannel);
    
23.             p = CallAIDLL.PredictGrayImagePath2Intptr(ImagePath, sliceImageSize, overRate, image_threshold, classnum, GPUindex);
    
24.             HObject outputImage = null;
    
25.             HOperatorSet.GenEmptyObj(out outputImage);
    
26.             outputImage.Dispose();
    
27.             HOperatorSet.GenImage1Extern(out outputImage, "byte", Width, Height, p, Marshal.GetFunctionPointerForDelegate(callback));
    
28.             string savepath2 = System.Windows.Forms.Application.StartupPath + @"/" + "gray_2.png";
    
29.             HOperatorSet.WriteImage(outputImage, "png", -1, savepath2);
    
30. 
    
31.             // method3 -- halcon12进行调试
    
32.             HObject ho_Image1 = null;
    
33.             HTuple aWidth = 1200, aHeight = 1200;
    
34.             HOperatorSet.GenEmptyObj(out ho_Image1);
    
35.             ho_Image1.Dispose();
    
36.             HOperatorSet.ReadImage(out ho_Image1, ImagePath);
    
37.             HOperatorSet.GetImageSize(ho_Image1, out aWidth, out aHeight);
    
38.             // 将图像转化为bytes传入
    
39.             HTuple hred, hgreen, hblue, type0, width0, height0;
    
40.             HOperatorSet.GetImagePointer1(ho_Image1, out hred, out type0, out width0, out height0);
    
41.             //HOperatorSet.GetImagePointer3(ho_Image1, out hred, out hgreen, out hblue, out type0, out width0, out height0);
    
42.             byte[] bptr = new byte[aWidth.I * aHeight.I * ImageChannel];
    
43.             for (int i = 0; i < (width0.I * height0.I); i++)
    
44.             {
    
45.                 Marshal.Copy((IntPtr)(hred + i), bptr, i * 1 + 0, 1);
    
46.             }
    
47.             IntPtr p0 = Marshal.AllocHGlobal(aWidth.I * aHeight.I * ImageChannel);
    
48.             Marshal.Copy(bptr, 0, p0, aWidth.I * aHeight.I * ImageChannel);
    
49.             // call
    
50.             IntPtr p1 = Marshal.AllocHGlobal(aWidth.I * aHeight.I * ImageChannel);
    
51.             p1 = CallAIDLL.PredictGrayImageMat2Intptr(aWidth.I, aHeight.I, ImageChannel, p0, sliceImageSize, overRate, image_threshold, classnum, GPUindex);
    
52.             HObject outputImage1 = null;
    
53.             HOperatorSet.GenEmptyObj(out outputImage1);
    
54.             outputImage1.Dispose();
    
55.             HOperatorSet.GenImage1Extern(out outputImage1, "byte", aWidth.I, aHeight.I, p1, Marshal.GetFunctionPointerForDelegate(callback));
    
56.             string savepath3 = System.Windows.Forms.Application.StartupPath + @"/" + "gray_3.png";
    
57.             HOperatorSet.WriteImage(outputImage1, "png", -1, savepath3);
    
58. 
    
59.             // method4 -- halcon12进行调试
    
60.             string savepath4 = System.Windows.Forms.Application.StartupPath + @"/" + "gray_4.png";
    
61.             HObject ho_Image2 = null;
    
62.             HTuple aWidth1 = 1200, aHeight1 = 1200;
    
63.             HOperatorSet.GenEmptyObj(out ho_Image2);
    
64.             ho_Image2.Dispose();
    
65.             HOperatorSet.ReadImage(out ho_Image2, ImagePath);
    
66.             HOperatorSet.GetImageSize(ho_Image2, out aWidth1, out aHeight1);
    
67.             // 将图像转化为bytes传入
    
68.             HTuple hred1, hgreen1, hblue1, type1, width1, height1;
    
69.             HOperatorSet.GetImagePointer1(ho_Image2, out hred1, out type1, out width1, out height1);
    
70.             //HOperatorSet.GetImagePointer3(ho_Image2, out hred1, out hgreen1, out hblue1, out type1, out width1, out height1);
    
71.             byte[] bptr1 = new byte[aWidth1.I * aHeight1.I * ImageChannel];
    
72.             for (int i = 0; i < (aWidth1.I * aHeight1.I); i++)
    
73.             {
    
74.                 Marshal.Copy((IntPtr)(hred1 + i), bptr1, i * 1 + 0, 1);
    
75.                 //Marshal.Copy((IntPtr)(hgreen1 + i), bptr1, i * 3 + 1, 1);
    
76.                 //Marshal.Copy((IntPtr)(hred1 + i), bptr1, i * 3 + 2, 1);
    
77.             }
    
78.             IntPtr p2 = Marshal.AllocHGlobal(aWidth1.I * aHeight1.I * ImageChannel);
    
79.             Marshal.Copy(bptr1, 0, p2, aWidth1.I * aHeight1.I * ImageChannel);
    
80.             // call
    
81.             CallAIDLL.PredictGrayImageMat2Path(aWidth1, aHeight1, ImageChannel, p2, savepath4, sliceImageSize, overRate, image_threshold, classnum, GPUindex);
    
82. 
    
83.             //
    
84.             MessageBox.Show("测试完成！！！");

复制代码

witiai

彩色图像：

1.             string ImagePath = this.textBox2.Text.ToString();
   
2.             int sliceImageSize = 256;
   
3.             double overRate = 0.0;
   
4.             double image_threshold = 0.5;
   
5.             int classnum = 4;
   
6.             int GPUindex = 0;
   
7.             int ImageChannel = 3;
   
8.             string savepath1 = System.Windows.Forms.Application.StartupPath + @"/" + "rgb_1.png";
   
9.             // 初始化AI模型
   
10.             string AImodelpt = @"rgb_model.WITIAI";
    
11.             // method1
    
12.             CallAIDLL.load_module(AImodelpt, GPUindex);
    
13.             CallAIDLL.PredictRGBImage(ImagePath, savepath1, sliceImageSize, overRate, image_threshold, classnum, GPUindex);
    
14. 
    
15.             // method2 -- halcon12进行调试
    
16.             HObject ho_Image = null;
    
17.             HTuple Width = 1200, Height = 1200;
    
18.             HOperatorSet.GenEmptyObj(out ho_Image);
    
19.             ho_Image.Dispose();
    
20.             HOperatorSet.ReadImage(out ho_Image, ImagePath);
    
21.             HOperatorSet.GetImageSize(ho_Image, out Width, out Height);
    
22.             IntPtr p = Marshal.AllocHGlobal(Width.I * Height.I * ImageChannel);
    
23.             p = CallAIDLL.PredictRGBImagePath2Intptr(ImagePath, sliceImageSize, overRate, image_threshold, classnum, GPUindex);
    
24.             HObject outputImage = null;
    
25.             HOperatorSet.GenEmptyObj(out outputImage);
    
26.             outputImage.Dispose();
    
27.             HOperatorSet.GenImage1Extern(out outputImage, "byte", Width, Height, p, Marshal.GetFunctionPointerForDelegate(callback));
    
28.             string savepath2 = System.Windows.Forms.Application.StartupPath + @"/" + "rgb_2.png";
    
29.             HOperatorSet.WriteImage(outputImage, "png", -1, savepath2);
    
30. 
    
31.             // method3 -- halcon12进行调试
    
32.             HObject ho_Image1 = null;
    
33.             HTuple aWidth = 1200, aHeight = 1200;
    
34.             HOperatorSet.GenEmptyObj(out ho_Image1);
    
35.             ho_Image1.Dispose();
    
36.             HOperatorSet.ReadImage(out ho_Image1, ImagePath);
    
37.             HOperatorSet.GetImageSize(ho_Image1, out aWidth, out aHeight);
    
38.             // 将图像转化为bytes传入
    
39.             HTuple hred, hgreen, hblue, type0, width0, height0;
    
40.             HOperatorSet.GetImagePointer3(ho_Image1, out hred, out hgreen, out hblue, out type0, out width0, out height0);
    
41.             byte[] bptr = new byte[aWidth.I * aHeight.I * ImageChannel];
    
42.             for (int i = 0; i < (width0.I * height0.I); i++)
    
43.             {
    
44.                 Marshal.Copy((IntPtr)(hblue + i), bptr, i * 3 + 0, 1);
    
45.                 Marshal.Copy((IntPtr)(hgreen + i), bptr, i * 3 + 1, 1);
    
46.                 Marshal.Copy((IntPtr)(hred + i), bptr, i * 3 + 2, 1);
    
47.             }
    
48.             IntPtr p0 = Marshal.AllocHGlobal(aWidth.I * aHeight.I * ImageChannel);
    
49.             Marshal.Copy(bptr, 0, p0, aWidth.I * aHeight.I * ImageChannel);
    
50.             // call
    
51.             IntPtr p1 = Marshal.AllocHGlobal(aWidth.I * aHeight.I * ImageChannel);
    
52.             p1 = CallAIDLL.PredictRGBImageMat2Intptr(aWidth.I, aHeight.I, ImageChannel, p0, sliceImageSize, overRate, image_threshold, classnum, GPUindex);
    
53.             HObject outputImage1 = null;
    
54.             HOperatorSet.GenEmptyObj(out outputImage1);
    
55.             outputImage1.Dispose();
    
56.             HOperatorSet.GenImage1Extern(out outputImage1, "byte", aWidth.I, aHeight.I, p1, Marshal.GetFunctionPointerForDelegate(callback));
    
57.             string savepath3 = System.Windows.Forms.Application.StartupPath + @"/" + "rgb_3.png";
    
58.             HOperatorSet.WriteImage(outputImage1, "png", -1, savepath3);
    
59. 
    
60.             // method4 -- halcon12进行调试
    
61.             string savepath4 = System.Windows.Forms.Application.StartupPath + @"/" + "rgb_4.png";
    
62.             HObject ho_Image2 = null;
    
63.             HTuple aWidth1 = 1200, aHeight1 = 1200;
    
64.             HOperatorSet.GenEmptyObj(out ho_Image2);
    
65.             ho_Image2.Dispose();
    
66.             HOperatorSet.ReadImage(out ho_Image2, ImagePath);
    
67.             HOperatorSet.GetImageSize(ho_Image2, out aWidth1, out aHeight1);
    
68.             // 将图像转化为bytes传入
    
69.             HTuple hred1, hgreen1, hblue1, type1, width1, height1;
    
70.             HOperatorSet.GetImagePointer3(ho_Image2, out hred1, out hgreen1, out hblue1, out type1, out width1, out height1);
    
71.             byte[] bptr1 = new byte[aWidth1.I * aHeight1.I * ImageChannel];
    
72.             for (int i = 0; i < (aWidth1.I * aHeight1.I); i++)
    
73.             {
    
74.                 Marshal.Copy((IntPtr)(hblue1 + i), bptr1, i * 3 + 0, 1);
    
75.                 Marshal.Copy((IntPtr)(hgreen1 + i), bptr1, i * 3 + 1, 1);
    
76.                 Marshal.Copy((IntPtr)(hred1 + i), bptr1, i * 3 + 2, 1);
    
77.             }
    
78.             IntPtr p2 = Marshal.AllocHGlobal(aWidth1.I * aHeight1.I * ImageChannel);
    
79.             Marshal.Copy(bptr1, 0, p2, aWidth1.I * aHeight1.I * ImageChannel);
    
80.             // call
    
81.             CallAIDLL.PredictRGBImageMat2Path(aWidth1, aHeight1, ImageChannel, p2, savepath4, sliceImageSize, overRate, image_threshold, classnum, GPUindex);
    
82. 
    
83.             //
    
84.             MessageBox.Show("OK");

复制代码

witiai

模型释放：

1.             string AImodelpt = @"rgb_model.WITIAI";
   
2.             CallAIDLL1.Free_module(AImodelpt, 0);

复制代码