Более чистый способ отбеливания каждого изображения в партии с использованием керас

Question

Я хотел бы отбелить каждое изображение в серии.Код, который я должен сделать, таков:

def whiten(self, x):
    shape = x.shape
    x = K.batch_flatten(x)
    mn = K.mean(x, 0)
    std = K.std(x, 0) + K.epsilon()
    r = (x - mn) / std
    r = K.reshape(x, (-1,shape[1],shape[2],shape[3]))
    return r
#

, где x - это (?, 320,320,1).Я не заинтересован в функции изменения формы с аргументом -1.Есть ли более чистый способ сделать это?

Answer 1

Давайте посмотрим, что делает -1.Из документации Tensorflow (поскольку документация от Keras недостаточна по сравнению с документацией из Tensorflow):

Если один компонент формы имеет специальное значение -1, размер этого измерения вычисляется так, чтобыобщий размер остается постоянным.

Итак, что это значит:

from keras import backend as K

X = tf.constant([1,2,3,4,5])
K.reshape(X, [-1, 5])
# Add one more dimension, the number of columns should be 5, and keep the number of elements to be constant
# [[1 2 3 4 5]]

X = tf.constant([1,2,3,4,5,6])
K.reshape(X, [-1, 3])
# Add one more dimension, the number of columns should be 3
# For the number of elements to be constant the number of rows should be 2
# [[1 2 3]
#  [4 5 6]]

Я думаю, что это достаточно просто.Итак, что происходит в вашем коде:

# Let's assume we have 5 images, 320x320 with 3 channels
X = tf.ones((5, 320, 320, 3))
shape = X.shape

# Let's flat the tensor so we can perform the rest of the computation
flatten = K.batch_flatten(X)
# What this did is: Turn a nD tensor into a 2D tensor with same 0th dimension. (Taken from the documentation directly, let's see that below)
flatten.shape
# (5, 307200)
# So all the other elements were squeezed in 1 dimension while keeping the batch_size the same

# ...The rest of the stuff in your code is executed here...

# So we did all we wanted and now we want to revert the tensor in the shape it had previously
r = K.reshape(flatten, (-1, shape[1],shape[2],shape[3]))
r.shape
# (5, 320, 320, 3)

Кроме того, я не могу придумать более чистого способа сделать то, что вы хотите сделать.Если вы спросите меня, ваш код уже достаточно ясен.