3

问题

我正在寻找可能已经存在的 repa 库中的函数。我想要一个功能：

  1. 采用二维数组
  2. 指定窗口大小的两个整数
  3. 在二维数组上给定大小的每个窗口中，计算一个新值，例如该特定窗口中的小值。

例子

min使用 3x3 窗口映射函数：

| 1 2 3 4 3
  4 5 6 7 2
  7 8 9 4 2
  5 4 8 1 6
  8 5 3 3 2 |

将返回：

| 1 1 2 2 2
  1 1 2 2 2
  4 4 1 1 1
  4 3 1 1 1
  4 3 1 1 1 |

请注意，我正在使用类似于BoundClamp  构造函数的方案Data.Array.Repa.Stencil。这不是模板卷积，即它没有将模板应用于 2D 数组的每个元素。相反，它在数组的每个窗口上执行一个函数，边缘的超出范围的元素被分配到二维数组边缘的最接近的值。

可能解决方案的类型

该函数可能类似于：

mapF
  :: Source r a
  => Boundary a            -- ^ How to handle the boundary of the array.
  -> (Int,Int)             -- ^ window size in the X and Y direction.
  -> (Array r DIM2 a -> b) -- ^ function over window e.g. to return the minimum value.
  -> Array r DIM2 a        -- ^ Array to apply function to.
  -> Array r DIM2 b

这是已经存在的东西，还是编码起来很简单？

4

1 回答 1

1

我对我的 repa 生疏了，但相信你可以使用traverse并手动检测数组边界。考虑类型：

traverse ::
  (Source r a, Shape sh, Shape sh') =>
  Array r sh a
  -> (sh -> sh') -> ((sh -> a) -> sh' -> b) -> Array D sh' b

该函数采用原始数组，一个产生新形状的函数，一个采用查找函数和索引以产生新值的函数，并产生新的（延迟的）数组。

一个简单的解决方案是检查索引的所有邻居，使用minand控制边界max：

import qualified Data.Array.Repa as R
import           Data.Array.Repa (Z(..), traverse, fromListUnboxed, toList, (:.)(..))
import Prelude
import Data.List.Split

main = do let a = fromListUnboxed (Z :. h :. w) ( [1..6] ++ [2..7] ++ [3..8] ++ [4..9] ++ [5..10] ::  [Int])
              r = traverse a id (\f (Z :. y :. x) -> minimum [f (Z :. yi :. xi) | xi <- idx w x, yi <- idx h y])
          printArray a
          printArray r
  where
    idx b i = map (bound b) [i, i+1, i-1]
    bound b = min (b-1) . max 0
    w = 6 :: Int
    h = 5 :: Int

    printArray = putStrLn . unlines . map show . chunksOf w . toList

反对此解决方案的主要原因是性能（相同数字的许多比较，应静态消除的许多边界检查）。OTOH，您的问题只要求一个简单的解决方案，并且似乎并没有过度关注性能。

如果 Repa 内置了更快的解决方案，我也很感兴趣。

于 2014-05-12T18:32:12.543 回答