satsim.dataset package

Submodules

satsim.dataset.augment module

satsim.dataset.augment.augment_satnet_with_satsim(dataset, augment_satsim_params, prob=0.5, rn=0, min_snr=2.0, box_pad=10)

Augments a SatNet Dataset with SatSim images. The SatSim image is added onto the example image. New synthetic targets are appended to the bounding box annotation list.

Parameters:
  • datasettf.Dataset, Dataset object that returns SatNet types: image: tf.float32, bounding box annotations: tf.float32, filename: tf.string, annotation filename : tf.string

  • augment_satsim_paramsdict, SatSim simulation parameters

  • probfloat, probability of augmenting the example image. default=0.5

  • rnfloat, estimated read noise of sensor in photoelectrons used to estimate SNR. default=0

  • min_snrfloat, minimum SNR to include synthetic targets in annotation list. default=2.0

  • box_padint, number of pixels to pad synthetic targets on each side. default=10

Returns:

A tf.Dataset, the mapped Dataset with SatSim augmentation