モルフォロジーのオープニングを返す入力

Question

私はモルフォロジーのオープニングをプログラミングしており、入力画像と同じ結果を返しています。私はそれの説明を誤解していますか？私のコードは次のとおりです。

Var x, y; 
Func limit, erosion, dilation; 

ImageParam input(type_of<uint8_t>(), 2); 
Param<int> dimension; 

RDom r(-1 * dimension/2, dimension, -1 * dimension/2, dimension); 
limit = BoundaryConditions::repeat_edge(input); 

erosion(x, y) = argmin(r, limit(x + r.x, y + r.y), "erosion")[2]; 
dilation(x, y) = argmax(r, erosion(x + r.x, y + r.y), "dilation")[2]; 

erosion.compute_root(); 
dilation.vectorize(x, 4).parallel(y); 

Target target = get_host_target(); 

target.set_feature(Target::NoRuntime, true); 

dilation.compile_to_static_library(path, { input,dimension }, target); 

Answer 1

次は私のためのOpenCVのチュートリアルの画像に類似した結果を生成するようだ：

#include "Halide.h" 
#include "../../tools/halide_image_io.h" 

using namespace Halide; 

int main(int argc, char **argv) { 
    Var x, y; 
    Func limit, erosion, dilation; 

    ImageParam input(type_of<uint8_t>(), 2); 
    Param<int> dimension; 

    RDom r(-1 * dimension/2, dimension, -1 * dimension/2, dimension); 
    limit = BoundaryConditions::repeat_edge(input); 

#if 0 
    erosion(x, y) = minimum(limit(x + r.x, y + r.y), "erosion"); 
    dilation(x, y) = maximum(erosion(x + r.x, y + r.y), "dilation"); 
#else 
    erosion(x, y) = argmin(r, limit(x + r.x, y + r.y), "erosion")[2]; 
    dilation(x, y) = argmax(r, erosion(x + r.x, y + r.y), "dilation")[2]; 
#endif 

    erosion.compute_root(); 
    dilation.vectorize(x, 4).parallel(y); 

    Image<uint8_t> in = Halide::Tools::load_image("/some/path/morphology.png"); 
    input.set(in); 
    dimension.set(10); 

    Image<uint8_t> result = dilation.realize(in.width(), in.height()); 

    Halide::Tools::save_image(result, "/some/path/morphology_out.png"); 

    return 0; 
} 

の#if 0ブロックがだけではなく、最小値と最大演算子を使用して示すことを意図していますargmin/argmaxのまた、argmin/argmaxのr引数はオプションです。

イメージがHalide AOTコンパイルされた関数にどのように渡されているか、または返された出力をどのように扱っているのかがわかります。また、debug_to_fileを使用するか、Halideコードで印刷して、パイプラインの中間段階をデバッグすることもできます。