Выполнить среднее вычитание в списке изображений?

Question

У меня есть массивный массив формы (100,320,320), 100 изображений, и каждое изображение имеет размер 320 * 320.

Я пытался сделать:

mean = np.mean(train_x)
train_x -= mean

Я получаю Cannot cast ufunc subtract output from dtype('float64') to dtype('uint8') with casting rule 'same_kind'

Может кто-нибудь, пожалуйста, подскажите мне, как этого добиться?

Answer 1

Либо разрешите результат быть числом с плавающей запятой:

mean = np.mean(train_x)
train_x = train_x - mean

Или рассчитайте среднее значение как np.uint8 (что теряет точность):

mean = np.mean(train_x, dtype=train_x.dtype)
train_x -= mean