Je souhaite implémenter la normalisation d'instance

Question

J'écris un code cnn métallique. Metal fournit MPSCNNLocalContrastNormalization, Étant donné que le concept de normalisation d'instance est légèrement différent, j'ai l'intention de l'implémenter en tant que fonction noyau.

Cependant, le problème est que la moyenne et la variance pour chaque R, G, B devraient être obtenues lorsque la caractéristique est R, G, B dans la texture reçue de l'entrée dans la fonction du noyau. Je veux obtenir quelques conseils sur la façon de mettre en œuvre cela.

kernel void instance_normalization_2darray(texture2d_array<float, access::sample> src [[ texture(0) ]], 
             texture2d_array<float, access::write> dst [[ texture(1) ]], 
             uint3 tid [[thread_position_in_grid]]) { 

} 


    kernel void calculate_avgA(texture2d_array<float, access::read> texture_in [[texture(0)]], 
          texture2d_array<float, access::write> texture_out [[texture(1)]], 
          uint3 tid [[thread_position_in_grid]]) 
{ 
    int width = texture_in.get_width(); 
    int height = texture_in.get_height(); 
    int depth = texture_in.get_array_size(); 
    float4 outColor; 


    uint3 kernelIndex(0,0,0); 
    uint3 textureIndex(0,0,0); 

    for(int k = 0; k < depth; k++) { 
     outColor = (0.0, 0.0, 0.0, 0.0); 
     for (int i=0; i < width; i++) 
     { 
      for (int j=0; j < height; j++) 
      { 
       kernelIndex = uint3(i, j, k); 
       textureIndex = uint3(tid.x + i, tid.y + j, tid.z + k); 
       float4 color = texture_in.read(textureIndex.xy, textureIndex.z).rgba; 
       outColor += color; 
      } 
     } 
     outColor = outColor/(width * height); 
     texture_out.write(float4(outColor.rgba), tid.xy, textureIndex.z); 
    } 
} 

Answer 1

Mr.Bista J'ai eu le même problème pour cela, Apple n'a pas fourni une fonction pour cela avec vitesse rapide. Et j'utilise juste MPSCNNPoolingAverage pour la moyenne de caculate avant les noyaux. Peut-être que c'est une méthode temporaire pour cela. Et l'autre algorithme n'est pas mieux que cela, comme l'algorithme de somme de réduction après mon test avec des codes. Je vais donc continuer à suivre une meilleure implémentation pour cela.