Halide模板匹配

Halcom

Halide模板匹配

1. // windows-20240425
   
2. // WITIAITemplateMatching_generate -o . -g TemplateWitiai -f TemplateWitiai -e static_library,h,schedule target=host
   
3. // WITIAITemplateMatching_generate -o . -g TemplateWitiai -f TemplateWitiai -e static_library,h,schedule -p autoschedule_mullapudi2016.dll target=host autoscheduler=Mullapudi2016 autoscheduler.parallelism=32 autoscheduler.last_level_cache_size=16777216 autoscheduler.balance=40
   
4. #include "Halide.h"
   
5. #include <stdio.h>
   
6. using namespace Halide;
   
7. // We will define a generator to auto-schedule.
   
8. class TemplateWitiaiMain : public Halide::Generator<TemplateWitiaiMain> {
   
9. public:
   
10.         Input<Buffer<uint8_t, 3>> input1{ "input1" };
    
11.         Input<Buffer<uint8_t, 3>> templ{ "templ" };
    
12.         Output<Buffer<float, 2>> output1{ "output1" };
    
13.         Output<Buffer<float, 2>> output2{ "output2" };
    
14.         void generate()
    
15.         {
    
16.                 //input1.dim(0).set_stride(3);  // stride in dimension 0 (x) is three
    
17.                 //input1.dim(1).set_stride(3);  // stride in dimension 1 (y) is three
    
18.                 input1.dim(0).set_bounds(0, input1.width());
    
19.                 input1.dim(1).set_bounds(0, input1.height());
    
20.                 //input1_1(x, y) = input1(x, y, 0);
    
21.                 //input_16(x, y) = cast<uint16_t>(input1_1(x, y));
    
22.                 //limit = BoundaryConditions::constant_exterior(input_16, 0, 0, input1.width(), 0, input1.height());
    
23.                 limit = BoundaryConditions::repeat_edge(input1);
    
24.                 input1_1(x, y) = limit(x, y, 0);
    
25.                 input_16(x, y) = cast<float>(input1_1(x, y));
    
26.                 templ_1(x, y) = templ(x, y, 0);
    
27.                 temp_16(x, y) = cast<float>(templ_1(x, y));
    
28.                 RDom matchDom(0, templ.width(), 0, templ.height());
    
29.                 //matchDom.where(matchDom.x + matchDom.y > 25);  // 增大步长
    
30.                 //Expr strided_x = x * 2;
    
31.                 //Expr strided_y = x * 2;
    
32.                 score(x, y) = sum(matchDom, pow(temp_16(matchDom.x, matchDom.y) - input_16(x * 3 + matchDom.x, y * 3 + matchDom.y), 2));
    
33.                 //
    
34.                 RDom searchDom(0, input1.width() / 3 - templ.width(), 0, input1.height() / 3 - templ.height());
    
35.                 Tuple searchBest = argmin(searchDom, score(searchDom.x, searchDom.y), "argmin");
    
36.                 //searchBest = argmin(searchDom, score(searchDom.x, searchDom.y), "argmin");
    
37.                 //Func bestX;
    
38.                 bestX(x, y) = searchBest[0];
    
39.                 //Func bestY;
    
40.                 bestY(x, y) = searchBest[1];
    
41.                 //Realization re = best.realize();
    
42.                 //Buffer<int> x_coordinate(re[0]);
    
43.                 //Buffer<int> y_coordinate(re[1]);
    
44.                 //int bestX0 = bestX(0);
    
45.                 //int bestY0 = bestY(0);
    
46. 
    
47.                 //output1(x, y) = cast<uint8_t>(best(x, y));
    
48.                 output1(x, y) = cast<float>(bestX(x, y));
    
49.                 output2(x, y) = cast<float>(bestY(x, y));
    
50.         }
    
51. 
    
52.         void schedule()
    
53.         {
    
54.                 if (using_autoscheduler())
    
55.                 {
    
56.                         //
    
57.                 }
    
58.                 else
    
59.                 {
    
60.                         limit.compute_root();
    
61.                         input1_1.compute_root();
    
62.                         input_16.compute_root();
    
63.                         templ_1.compute_root();
    
64.                         temp_16.compute_root();
    
65.                         score.vectorize(x, 8).parallel(y).compute_root();
    
66.                         //score.compute_root();
    
67.                         /*bestX.compute_root();
    
68.                         bestY.compute_root();*/
    
69.                         /*                bestX.parallel(x).parallel(y).compute_root();
    
70.                                         bestY.parallel(x).parallel(y).compute_root();*/
    
71.                                         /*bestX.vectorize(x, 16).parallel(y).compute_root();
    
72.                                         bestY.vectorize(x, 16).parallel(y).compute_root();*/
    
73.                         bestX.vectorize(x, 16).parallel(y).compute_root();
    
74.                         bestY.vectorize(x, 16).parallel(y).compute_root();
    
75.                         //bestX.tile(x, y, x_outer, y_outer, x_inner, y_inner, 64, 64)
    
76.                         //        .fuse(x_outer, y_outer, tile_index)
    
77.                         //        .vectorize(x_inner, 4)
    
78.                         //        .parallel(tile_index)
    
79.                         //        .compute_root();
    
80.                         //bestY.tile(x, y, x_outer, y_outer, x_inner, y_inner, 64, 64)
    
81.                         //        .fuse(x_outer, y_outer, tile_index)
    
82.                         //        .vectorize(x_inner, 4)
    
83.                         //        .parallel(tile_index)
    
84.                         //        .compute_root();
    
85.         /*                output1.vectorize(x, 16).parallel(y).compute_root();
    
86.                         output2.vectorize(x, 16).parallel(y).compute_root();*/
    
87.                         output1.compute_root();
    
88.                         output2.compute_root();
    
89.                 }
    
90.         }
    
91. 
    
92. private:
    
93.         Var x{ "x" }, y{ "y" }, c{ "c" }, x_outer{ "x_outer" }, y_outer{ "y_outer" }, x_inner{ "x_inner" }, y_inner{ "y_inner" }, tile_index{ "tile_index" };
    
94.         Func limit, input_16, input1_1, templ_1, temp_16, score, bestX, bestY;
    
95.         //Tuple searchBest;
    
96.         //RDom matchDom, searchDom;
    
97. };
    
98. 
    
99. // file along with tools/GenGen.cpp.
    
100. HALIDE_REGISTER_GENERATOR(TemplateWitiaiMain, TemplateWitiai)

复制代码