优化像素处理循环

Question

我正在实现一个算法（在 OpenCV 中），它迭代图像中的每个像素，并为每个像素计算与邻域中的像素匹配的块，以评估这些相邻像素的相似性。具有非常深循环的“幼稚”实现非常慢，所以我想知道如何尝试提高性能。以下是我当前代码的摘录：

for(nCh=1;nCh<=channels;nCh++) { // Loop over three channels

  for(i=0;i<h;i++) { // "vertical" loop

    for(j=0;j<w;j++) { // "horizontal" loop

      for (si=-sw_height; si<sw_height; si++){ // vertical search window loop

        for (sj=-sw_width; sj<sw_width; sj++){ // horizontal search window loop

            dist = 0;

            for (blki=0; blki<blk_height; blki++){ // block match loop

              for (blkj=0; blkj<blk_width; blkj++){ // block match loop

                current_pxl = data[(i+blki)*step+(j+blkj)*channels+nCh];  

                search_pxl  = data[(i+blki+si)*step+(j+blkj+sj)*channels+nCh]; 

                dist += pow((current_pxl - search_pxl),2); 

              }

            }

          // ... further processing

        }

      }

    }

  }

}

Answer 1

您在最里面的循环中调用pow。不要。

您还在那里进行了大量的索引计算。 我敢打赌，您可以将一些内容移出内部循环。

你应该能够得到它，所以你的内部循环看起来更像这样：

          for (blkj = 0; blkj < blk_width; blkj++, pc += channels, ps += channels){
            int diff = (*pc - *ps); 
            dist += (diff * diff); 
          }

然后，您甚至可能想要展开它。

顺便说一句，多一点空格可能会有所帮助:-)