Медианный фильтр с RenderScript на Android - PullRequest
Новая
       56

Медианный фильтр с RenderScript на Android

0 голосов
/ 30 января 2020

Я делаю первые шаги в обработке изображений с помощью RenderScript. Сейчас я пытаюсь реализовать медианный фильтр для Android с ним. В основном я пытаюсь сделать следующее: он получает изображение в растровом формате, запускает фильтр 3x3 для каждого пикселя с его 8 соседями (без краевых случаев), вычисляет медиану для каждого цветового канала и соответственно устанавливает выходной пиксель.

Я пробовал, если он работает с простым инвертированным рендерингом вроде этого: 

uchar4 RS_KERNEL invert(uchar4 in, uint32_t x, uint32_t y) {
  uchar4 out = in;
  out.r = 255 - in.r;
  out.g = 255 - in.g;
  out.b = 255 - in.b;
  return out;
}

, и он отлично работает. Но когда я изменяю RenderScript для вычисления медианы, я просто получаю вывод черного изображения.

Вот код RenderScript: 

#pragma version(1)
#pragma rs_fp_relaxed
#pragma rs java_package_name(com.google.android.GoogleCamera)

#include "rs_debug.rsh"

rs_allocation input;
uint32_t width;
uint32_t height;

static uchar medi(uchar colors[]) {
    //sorting just to half
    for(int i = 0; i < 5; i++){
        int minIndex = i;
        uchar minValue = colors[i];
        //loop until end
        for(int j = i+1; j < 9; j++){
            if (colors[j] < minValue) {
                minIndex = j;
                minValue = colors[j];
                //swap
                colors[j] = colors[i];
                colors[i] = minValue;
            }
        }
    }
    return colors[4];
}

uchar4 __attribute__((kernel)) median(uchar4 in, uint32_t x, uint32_t y) {
    uchar4 out = in;

    //edge
    if (x == 0 || x == width || y == 0 || y == height)
        return out;

    //all surronding pixels (TopLeft .. BottomRight)
    uchar4 tl = rsGetElementAt_uchar4(input, x-1, y+1);
    uchar4 tm = rsGetElementAt_uchar4(input, x, y+1);
    uchar4 tr = rsGetElementAt_uchar4(input, x+1, y+1);
    uchar4 l = rsGetElementAt_uchar4(input, x-1, y);
    uchar4 r = rsGetElementAt_uchar4(input, x+1, y);
    uchar4 bl = rsGetElementAt_uchar4(input, x-1, y-1);
    uchar4 bm = rsGetElementAt_uchar4(input, x, y-1);
    uchar4 br = rsGetElementAt_uchar4(input, x+1, y-1);

    //array for each color channel
    uchar reds[] = {tl.r, tm.r, tr.r, l.r, r.r, bl.r, bm.r, br.r, in.r};
    uchar greens[] = {tl.g, tm.g, tr.g, l.g, r.g, bl.g, bm.g, br.g, in.g};
    uchar blues[] = {tl.b, tm.b, tr.b, l.b, r.b, bl.b, bm.b, br.b, in.b};
    uchar alphas[] = {tl.a, tm.a, tr.a, l.a, r.a, bl.a, bm.a, br.a, in.a};

    //get median values
    out.r = medi(reds);
    out.g = medi(greens);
    out.b = medi(blues);
    out.a = medi(alphas);

    return out;
}

и на всякий случай вызывающая java part: 

private static Runnable runMedian(final Bitmap original, final File file) {
        return new Runnable() {
            @Override
            public void run() {
                //Create new bitmap
                Bitmap bitmap = original.copy(original.getConfig(), true);

                //Create renderscript
                RenderScript rs = RenderScript.create(CameraActivity.staticContext);

                //Create allocation from Bitmap
                Allocation allocationA = Allocation.createFromBitmap(rs, bitmap);

                //Create allocation with same type
                Allocation allocationB = Allocation.createTyped(rs, allocationA.getType());

                //Create script from rs file
                ScriptC_median medianScript = new ScriptC_median(rs);

                medianScript.set_input(allocationA);
                medianScript.set_width(original.getWidth());
                medianScript.set_height(original.getWidth());
                medianScript.forEach_median(allocationA, allocationB);

                //Copy script result into bitmap
                allocationB.copyTo(bitmap);

                //Destroy everything to free memory
                allocationA.destroy();
                allocationB.destroy();
                medianScript.destroy();
                rs.destroy();

                //write resulting bitmap to file
                writeFile(file, bitmap, "median");
            }
        };
    }

Я надеюсь, что кто-то, кто знает немного больше, может помочь мне. С наилучшими пожеланиями

1 Ответ

0 голосов
/ 05 февраля 2020

Возможно, вам потребуется пересмотреть обработку краев: индекс x изменяется от 0 до width-1, поэтому ребро имеет ширину-1 (и height-1).

В java part, medianScript.set_height(original.getWidth());, вероятно, должен читать original.getHeight(), если ширина и высота не совпадают.

...