Ошибка Dtype при использовании увеличения изображения в Imgaug

Question

Я использую Imgaug для увеличения изображений на моих тренировочных данных. Код выглядит следующим образом:

          training_augment = iaa.Sequential([iaa.Fliplr(0.5),
                                       iaa.Flipud(0.5),
                                       iaa.Sometimes(0.5, iaa.Rot90((1, 3))),
                                       iaa.Sometimes(0.25, iaa.Affine(rotate=(-30, 30),
                                                                      scale=(1.0, 1.25),
                                                                      translate_percent={"x": (-0.1, 0.1), "y": (-0.1, 0.1)},
                                                                      cval=1.0)),
                                       iaa.AdditiveGaussianNoise(loc=0, scale=(0.0, 0.2), per_channel=0.5)])

Следующая ошибка:

ValueError: получен dtype 'float64', который является запрещенным dtype (uint32, uint64, uint128, uint256, int32, int64, int128, int256, float64, float96, float128, float256).

Эта ошибка возникает при использовании AdditiveGuassianNoise, при использовании остальных он работает нормально. Я не совсем понимаю, что означает ошибка. Если бы кто-нибудь мог помочь, был бы очень признателен. Спасибо

Также добавлен мой код генератора пользовательских данных туда, куда идут изображения:

          def create_augmented_batch(self, index):  # call in the training images
        image_list, labels_batch = self._create_balanced_batch(index)
        if self.augmentation is not None:
            image_list = self.augmentation.augment(images=image_list)
        return image_list, labels_batch
    
    def __getitem__(self, index):
        if self.augmentation is None:
            images, labels = self.create_batch(index)  # for test images
        else:
            images, labels = self.create_augmented_batch(index)  # for training images
        return images[..., np.newaxis], tf.keras.utils.to_categorical(np.array(labels), num_classes=self.num_classes)