2feaa30964f4393ce65a394253e180cb8f6e0dc1,luminoth/datasets/object_detection_dataset.py,ObjectDetectionDataset,_augment,#ObjectDetectionDataset#,54

Before Change

            aug_fn = DATA_AUGMENTATION_STRATEGIES[aug_type]

            random_number = tf.random_uniform([])
            prob = aug_config.pop("prob", default_prob)
            apply_aug_strategy = tf.less(random_number, prob)

            augmented = tf.cond(
                apply_aug_strategy,
                lambda: aug_fn(image, bboxes, **aug_config),
                lambda: {"image": image, "bboxes": bboxes}

After Change

            aug_fn = DATA_AUGMENTATION_STRATEGIES[aug_type]

            random_number = tf.random_uniform([])
            prob = tf.to_float(aug_config.pop("prob", default_prob))
            apply_aug_strategy = tf.less(random_number, prob)

            augmented = tf.cond(
                apply_aug_strategy,
                lambda: aug_fn(image, aug_config, bboxes),
                lambda: {"image": image, "bboxes": bboxes}

Instances