0

C で記述されたノイズ除去アルゴリズムがあり、weave.inline を使用してコードを実行し、画像からノイズを除去する必要があります。アルゴリズムが正しく動作しません。Weave は 1D 配列に取り組んでいると書かれているため、2D 配列を 1D に変換する必要があります。それが私が行ったことですが、1つの1D配列をCアルゴリズムに送信すると、反復回数に応じてゼロのみが返されるか、送信した1Dとまったく同じように見えるものが得られるようです。 （しかし、それらが同じである場合、平等テストを受けると、そうではないことがわかります）。誰かが私のアルゴリズムの問​​題を理解できますか? コードは次のとおりです。

#!/usr/bin/env python
from PIL import Image
import numpy as np
from scipy.weave import inline
from scipy.weave import converters

def weave_iso_difusion_denoising(data0, m, n, kappa, iters):
    """
    Function of removing noise from pictures
   """

    c_code = r"""
     // Fill top and bottom edge
    for (int i=0; i<m; i++)
    {
        data1[m*i*0] = data0[m*i*0];
        data1[i*(n-1)] = data0[i*(n-1)];
    }

    // Fill left and right edges
    for (int i=1; i<n-1; i++) // skip corners
    {
        data1[n*0*i] = data0[n*0*i];
        data1[(n-1)*i] = data0[(n-1)*i];
    }

    for (int iter=0; iter<iters; iter++)
    {
        // Fill interior
        for (int i=1; i<m-1; i++)
        {
            for (int j=1; j<n-1; j++)
            {
                data1[n*i+j] = data0[n*i+j] +kappa*(data0[n*(i-1)+j]
                        +data0[n*i+(j-1)] -4*data0[n*i+j] +
                    data0[n*i+(j+1)] + data0[n*(i+1)+j]);
            }
        }
       tmp = data0;
       data0 = data1;
       data1 = tmp;
    }
    """
    data1 = np.zeros(data0.shape, dtype=data0.dtype)
    data1 = data1.ravel()
    tmp =  np.zeros(data0.shape, dtype=data0.dtype)
    tmp = tmp.ravel()
    var = ['data0', 'm', 'n', 'kappa', 'iters','data1','tmp']
    inline(c_code,var,type_converters=converters.blitz, compiler='gcc')
    return data1

if __name__ == '__main__':

    # from image-file to Numpy
    img = Image.open("disaster_before.jpg") # the noisy picture
    data0 = np.array(img.getdata()) # 1D-array
    print 'data0:',data0

    # executing the denoise-program
    kappa = 0.05; iters = 11; n = 375; m = 500
    data2 = weave_iso_difusion_denoising(data0, m, n,kappa, iters)
    print 'data:',data2

    print 'data2 - data0:',np.sum(abs(data2 - data0))

    # from Numpy to image-file
    img1 = Image.new('L', (m,n))
    img1.putdata(data2)
    img1.save("disaster_after_weave_v1.jpg")

output
"""
data0: [156 181 177 ..., 124  35  90]
data: [156 181 177 ..., 124  35  90]
data2 - data0: 18776776 # data0 and data1 is not the same even though they look the same
"""
4

1 に答える 1

0

この発言で

print 'data2 - data0:',np.sum(abs(data2 - data0))

data0とdata1が同じか、または異なるかをテストできます。ポイントは、アルゴリズムが機能するかどうかを確認することです。

于 2014-11-08T18:33:06.550 に答える