Comment calculer la variance unitaire dans le tenseur?

Question

Mon entrée est un ensemble d'images et je veux calculer l'univariance sur mes images. Mais quand essayer de vérifier avec numpy, la variance d'unité devrait donner 1 à la fin. Qu'est-ce que je fais de mal dans mon code?

def pre_processing(img_list, zero_mean=True, unit_var=True): 
    with tf.device('/cpu:0'): 
     tn_img0 = img_list[0][1] 
     tn_img1 = img_list[1][1] 

     t_img = tn_img0 
     # t_img = tf.concat([tn_img0, tn_img1], axis=0) 
     rgb_mean, rgb_var = tf.nn.moments(t_img, [0, 1]) 

     if zero_mean: 
      tn_img0 = tf.subtract(img_list[0][1], rgb_mean) 
      tn_img1 = tf.subtract(img_list[1][1], rgb_mean) 

     if unit_var: 
      tn_img0 = tf.divide(tn_img0, rgb_var) 
      tn_img1 = tf.divide(tn_img1, rgb_var) 

Answer 1

Vous devez diviser par le standard deviation pour obtenir une variance unitaire de vos entrées. Changez donc votre code pour:

tn_img0 = tf.divide(tn_img0, tf.sqrt(rgb_var)) 
tn_img1 = tf.divide(tn_img1, tf.sqrt(rgb_var))