yet another tail prepadding condition fix, fix #57

nihui · Feb 23, 2020 · 4b626f8 · 4b626f8
1 parent 1450ec6
commit 4b626f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/waifu2x.cpp b/src/waifu2x.cpp
@@ -188,7 +188,7 @@ int Waifu2x::process(const ncnn::Mat& inimage, ncnn::Mat& outimage) const
     for (int yi = 0; yi < ytiles; yi++)
     {
         int in_tile_y0 = std::max(yi * TILE_SIZE_Y - prepadding, 0);
-        int in_tile_y1 = std::min((yi + 1) * TILE_SIZE_Y + prepadding_bottom, h);
+        int in_tile_y1 = std::min((yi + 1) * TILE_SIZE_Y + (yi == ytiles - 1 ? prepadding_bottom : prepadding), h);
 
         ncnn::Mat in;
         if (net.opt.use_fp16_storage && net.opt.use_int8_storage)