Как умножить сверточное ядро ​​на 3х3 и изображение - PullRequest
0 голосов
/ 18 мая 2018

Имеется сверточное ядро ​​3x3 и изображение, представленное массивом пикселей целых значений.

Сверточное ядро ​​представляется следующим образом:

//compound convolutional kernels
//                                | 1, 0,  1|
// convolutional kernel H = src x | 0, 0,  0|
//                                |-1, 0, -1|

//                                | 1, 0, -1|
// convolutional kernel V = src x | 0, 0,  0|
//                                | 1, 0, -1|

сверточные ядра = ядроH + ядро ​​V

for(int inc=0; inc<height-2; inc++)
{
    //loaded 3 lines into memory
    str1_16pxs = _mm_loadu_si128((__m128i*)(src_all_str));
    str2_16pxs = _mm_loadu_si128((__m128i*)(src2_all_str));
    str3_16pxs = _mm_loadu_si128((__m128i*)(src3_all_str));

    //packing 16bit
    str1_16pxs_pack1st_8to16 = _mm_cvtepu8_epi16(str1_16pxs);
    str2_16pxs_pack1st_8to16 = _mm_cvtepu8_epi16(str2_16pxs);
    str3_16pxs_pack1st_8to16 = _mm_cvtepu8_epi16(str3_16pxs);

//---!
        //there is we make the first convolution for 8px's
        //... How ???
//---

    //summ 1st 8to16 vertical registers
    sum1_str12_vert_16pxs_pack1st_8to16  = _mm_add_epi16(str1_16pxs_pack1st_8to16,           str2_16pxs_pack1st_8to16);
    sum1_str123_vert_16pxs_pack1st_8to16 = _mm_add_epi16(sum1_str12_vert_16pxs_pack1st_8to16,str3_16pxs_pack1st_8to16);

    for(int jnc=0; jnc<(width >> 4); jnc++)
    {
        str1_16pxs_plus_8pxs = _mm_srli_si128(str1_16pxs, 8);
        str2_16pxs_plus_8pxs = _mm_srli_si128(str2_16pxs, 8);
        str3_16pxs_plus_8pxs = _mm_srli_si128(str3_16pxs, 8);

        //pack 2nd 8to16 registers (+8px's)
        str1_16pxs_pack2nd_8to16 = _mm_cvtepu8_epi16(str1_16pxs_plus_8pxs);
        str2_16pxs_pack2nd_8to16 = _mm_cvtepu8_epi16(str2_16pxs_plus_8pxs);
        str3_16pxs_pack2nd_8to16 = _mm_cvtepu8_epi16(str3_16pxs_plus_8pxs);

//---!
            //do convolution for the remaining 8px's and so on until the end of the read line
            //... How ???
//---

        //summ vertic 8to16 registers
        sum1_str12_vert_16pxs_pack2nd_8to16  = _mm_add_epi16(str1_16pxs_pack2nd_8to16,           str2_16pxs_pack2nd_8to16);
        sum1_str123_vert_16pxs_pack2nd_8to16 = _mm_add_epi16(sum1_str12_vert_16pxs_pack2nd_8to16,str3_16pxs_pack2nd_8to16);

//---!4     loading next 16 px's
        src_all_str += 16;
        src2_all_str += 16;
        src3_all_str += 16;

        //...

        _mm_store_si128((__m128i*)(dst_all_str), res);
        dst_all_str += 8;

    }//for(jnc)

}//for(inc)

1 Ответ

0 голосов
/ 07 июня 2018

Итак, пример кода:

void SSEcode_Conv3x3 (unsigned char *src, int width, int height, short *dst) 
{
// Assert that width is a multiple of 16
if (width & 0xF) return;

unsigned char* src_line1 = src;
unsigned char* src_line3 = src + 2 * width;

__m128i zero = _mm_setzero_si128();

for (int i = 0; i < height - 2; i++) 
{
    __m128i line1 = _mm_load_si128((__m128i*)src_line1);
    __m128i line3 = _mm_load_si128((__m128i*)src_line3);
     for (int j = 0; j < width / 16 - 1; j++)
     {
        src_line1 += 16;
        src_line3 += 16;

        __m128i line1next = _mm_load_si128((__m128i*)src_line1);
        __m128i line3next = _mm_load_si128((__m128i*)src_line3);

       //blablabla
#ifdef USE_CORE_H
_mm_add_epi16
_mm_add_epi16
_mm_sub_epi16
#endif
       //blablabla

       _mm_store_si128((__m128i*)(dst + 8), res);
       line1 = line1next;
       line3 = line3next;

       dst += 16;
     }//for (j)

     src_line1 += 16;
     src_line3 += 16;

     //blablabla

     _mm_store_si128((__m128i*)(dst + 8), res);
     dst += 16;
}//for (i)

}

Написание кода заняло много времени.Я новичок, так что жаль, что человек, который хорошо разбирается в СЕ, не помогал в интризах.

...