Module: fake_data_creator#
A class for creating fake data sets.
The FakeDataSet class allows the user to create fake data sets for testing, including generating images with random noise and adding artificial objects. The fake data can be saved to files or used directly.
- class kbmod.fake_data_creator.FakeDataSet(width, height, num_times, noise_level=2.0, psf_val=0.5, obs_per_day=3, use_seed=False)[source]#
This class creates fake data sets for testing and demo notebooks.
Methods
delete_fake_data
(data_dir)Remove the fake data in a given directory.
insert_object
(trj)Insert a fake object given the trajectory.
insert_random_object
(flux)Create a fake object and insert it into the image.
Make a stack of fake layered images.
save_fake_data
(data_dir)Create the fake data in a given directory.
save_time_file
(file_name)Save the mapping of visit ID -> timestamp to a file.
- delete_fake_data(data_dir)[source]#
Remove the fake data in a given directory.
- Parameters:
- data_dirstr
The path of the directory for the fake data.
- insert_object(trj)[source]#
Insert a fake object given the trajectory.
- Parameters:
- trjtrajectory
The trajectory of the fake object to insert.
- insert_random_object(flux)[source]#
Create a fake object and insert it into the image.
- Parameters:
- fluxfloat
The flux of the object.
- Returns:
- ttrajectory
The trajectory of the inserted object.