deepdisc.data_format.augment_image

Utilities for augmenting image data.

Attributes

LAMBDA_EFFS

A_EBV

Functions

redden(image[, rng_seed])

gaussblur(image[, rng_seed])

scale_psf(sigi, lambda_eff)

multiband_gaussblur(image[, rng_seed])

addelementwise16(image[, rng_seed])

addelementwise8(image[, rng_seed])

addelementwise(image[, rng_seed])

centercrop(image)

Crop an image to just the center portion

train_augs(image)

Get the augmentation list

dc2_train_augs(image)

Get the augmentation list

dc2_train_augs_full(image)

Get the augmentation list

hsc_test_augs(image)

Get the augmentation list

Module Contents

LAMBDA_EFFS = [3671, 4827, 6223, 7546, 8691, 9712][source]
A_EBV[source]
redden(image, rng_seed=None)[source]
Parameters:

  • image (ndarray)

  • rng_seed (np.random.Generator) – Random state that is seeded. if none, use machine entropy.

Return type:

augmented image

gaussblur(image, rng_seed=None)[source]
Parameters:

  • image (ndarray)

  • rng_seed (np.random.Generator) – Random state that is seeded. if none, use machine entropy.

Return type:

augmented image

scale_psf(sigi, lambda_eff)[source]
multiband_gaussblur(image, rng_seed=None)[source]
Parameters:

  • image (ndarray)

  • rng_seed (np.random.Generator) – Random state that is seeded. if none, use machine entropy.

Return type:

augmented image

addelementwise16(image, rng_seed=None)[source]
Parameters:

  • image (ndarray)

  • rng_seed (np.random.Generator) – Random state that is seeded. if none, use machine entropy.

Return type:

augmented image

addelementwise8(image, rng_seed=None)[source]
Parameters:

  • image (ndarray)

  • rng_seed (np.random.Generator) – Random state that is seeded. if none, use machine entropy.

Return type:

augmented image

addelementwise(image, rng_seed=None)[source]
Parameters:

  • image (ndarray)

  • rng_seed (np.random.Generator) – Random state that is seeded. if none, use machine entropy.

Return type:

augmented image

centercrop(image)[source]

Crop an image to just the center portion

Parameters:

image (ndarray)

Return type:

cropped image

train_augs(image)[source]

Get the augmentation list

Parameters:

image (image) – The image to be augmented

Returns:

augs – The list of augs for training. Set to RandomRotation, RandomFlip, RandomCrop

Return type:

detectron_addons.KRandomAugmentationList

dc2_train_augs(image)[source]

Get the augmentation list

Parameters:

image (image) – The image to be augmented

Returns:

augs – The list of augs for training. Set to RandomRotation, RandomFlip, RandomCrop

Return type:

detectron_addons.KRandomAugmentationList

dc2_train_augs_full(image)[source]

Get the augmentation list

Parameters:

image (image) – The image to be augmented

Returns:

augs – The list of augs for training. Set to RandomRotation, RandomFlip, RandomCrop

Return type:

detectron_addons.KRandomAugmentationList

hsc_test_augs(image)[source]

Get the augmentation list

Parameters:

image (image) – The image to be augmented

Returns:

augs – The augs for hsc testing. Set to 50% Crop due to memory constraints

Return type:

detectron2 AugmentationList