Use FITS catalogues, FITS image data and astropy to manipulate your data into a dynamically read data set!
See the example.ipynb notebook to see how you can easily load in a number of data sets into a dynamically memory loaded iterable data set.
With this package at ROOT_PATH/Cata2Data
on your deveice, install using pip:
pip install -e ROOT_PATH/Cata2Data
The quickest introcution:
from cata2data import CataData
from torch.utils.data import DataLoader
field_names = ["A", "B"]
catalogue_paths = ["CAT_A.fits", "CAT_A.fits"]
image_paths = ["IMG_A.fits", "IMG_B.fits"]
data = CataData(
catalogue_paths=catalogue_paths,
image_paths=image_paths,
field_names=field_names
)
dataloader = DataLoader(data, batch_size=64, shuffle=True)
See the doc strings for detailed notes on all of the parameters which CataData accepts. The main features are:
- Memmory mapping (
mmap
) - Image size for the loadable dataset can be set for the whole dataset, or
- Cutout sizes can be controlled by the catalog on a source by source basis, including for non uniform cutout heights and widths.
โ ๏ธ Note that currently catalogues are indexed through their"ra"
" and"dec"
columns. Use thecatalogue_preprocessing
parameter to correctly name the columns.
Open an issue and let us know what sort of issue you are experiencing.
Open a pull request if you have added functionality or fixed a bug.
CataData takes in fields of images and catalogues. Catalogues are merged into one dataframe and labelled with their respective field names. The length of CataData objects is the length of that dataframe. Entries are indexed through the dataframe and samples are cutout from the respective image using the units provided in the "RA" and "DEC" columns of the catalogue.
If catalogued features are needed to manipulate the iamges, we recommend using an image processing wrapper around CataData objects. I.e. a function like: image_postprocessing(catadata_instance, index) -> np.ndarray
which calls the catadata_object[index]
and manipulates the resulting image as required before returning it.