AM62A3: H3A Weight Problem for Each Partition

Hi enrui,

enrui zhang said:

I found that there was a parameter called uint8_t * weight, which should be the weight of each block in H3A

Yes, there is a weight input for AE.

enrui zhang said:

Coincidentally, I wanted to set different weights for each partition in H3A module

You may see how weight is applied to the H3A output in the code inside TI_AE_do as below.

    for (i = 0; i < height; i++)
    {
        for (j = 0; j < width; j++)
        {
            cnt_tol++;
            if (weight != NULL)
            {
                // 1024 * 64 * 64 * 128 = 2**29
                rsum += (int32_t)h3a_data[i * width + j].red   * weight[i * width + j];
                gsum += (int32_t)h3a_data[i * width + j].green * weight[i * width + j];
                bsum += (int32_t)h3a_data[i * width + j].blue  * weight[i * width + j];
                // 64 * 64 * 128 = 2**19
                wsum += weight[i * width + j];
            }
            else
            {
                rsum += (int32_t) h3a_data[i * width + j].red;
                gsum += (int32_t) h3a_data[i * width + j].green;
                bsum += (int32_t) h3a_data[i * width + j].blue;
                wsum += 1;
            }

enrui zhang said:

is there a convenient interface for setting weights?

You may simply make your own spatial weighting array as input to TI_AE_do.