メモリ周りの補助クラスを追加

[cuda.git] / binarize / binarize.cpp

diff --git a/binarize/binarize.cpp b/binarize/binarize.cpp
index 940132e..792b290 100644 (file)
--- a/binarize/binarize.cpp
+++ b/binarize/binarize.cpp
@@ -31,11 +31,19 @@ static void binarize_cpu(ImageGray& out, const ImageGray& in, const uint8_t thre
 
 int _tmain(int argc, _TCHAR* argv[])
 {
-       const uint8_t thres(128);
-       ImageGray image;
-
        try {
+               const uint8_t thres(128);
+               ImageGray image;
+
                image = loadPGM("..\\img\\sine.pgm");
+
+               ImageGray out(image.width(), image.height());
+               binarize_cpu(out, image, thres);
+               savePNM(out, "result_cpu.pgm");
+
+               ImageGray out_gpu(image.width(), image.height());
+               binarize_gpu(out_gpu, image, thres);
+               savePNM(out_gpu, "result_gpu.pgm");
        }
        catch (std::exception& e)
        {
@@ -43,14 +51,6 @@ int _tmain(int argc, _TCHAR* argv[])
                return -1;
        }
 
-       ImageGray out(image.width(), image.height());
-       binarize_cpu(out, image, thres);
-       savePNM(out, "result_cpu.pgm");
-
-       ImageGray out_gpu(image.width(), image.height());
-       binarize_gpu(out_gpu, image, thres);
-       savePNM(out_gpu, "result_gpu.pgm");
-
        return 0;
 }