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
dataset – tf.Dataset, Dataset object that returns SatNet types: image: tf.float32, bounding box nnnotations: tf.float32, filename: tf.string, annotation filename : tf.string
augment_satsim_params – dict, SatSim simulation parameters
prob – float, probability of augmenting the example image. default=0.5
rn – float, estimated read noise of sensor in photoelectrons used to estimate SNR. default=0
min_snr – float, minimum SNR to include synthetic targets in annotation list. default=2.0
box_pad – int, number of pixels to pad synthetic targets on each side. default=10
- Returns
A tf.Dataset, the mapped Dataset with SatSim augmentation