A PyTorch Dataset for the FITS file format
Project description
FITSDataset
This package contains a custom PyTorch Dataset for quick and easy training on FITS files, commonly used in astronomical data analysis. In particular, the FITSDataset class caches FITS files as PyTorch tensors for the purpose of increasing training speed.
Contributions and feedback are welcome; please open a pull request or an issue.
Quickstart
Using Python 3.6+, install from source with
python -m pip install git+https://github.com/amritrau/fitsdataset.git
Create a toy dataset with samples from the Hyper Suprime-Cam survey with:
>>> from fitsdataset import FITSDataset
>>> dataset = FITSDataset("path/to/examples/hsc/", size=101, label_col="target")
Notice that the cached tensors appear in path/to/examples/hsc/tensors.
Documentation
>>> from fitsdataset import FITSDataset
>>> help(FITSDataset)
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file fitsdataset-0.0.1.tar.gz.
File metadata
- Download URL: fitsdataset-0.0.1.tar.gz
- Upload date:
- Size: 3.2 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.2.0 pkginfo/1.5.0.1 requests/2.24.0 setuptools/50.3.0 requests-toolbelt/0.9.1 tqdm/4.48.2 CPython/3.8.1
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 7b5aaa7bd09448a1e4575d12d1a821e03a81261540de2bfb12f1776007708171 |
|
| MD5 | cebee4e7d28d6ab0fd0290d9cb563402 |
|
| BLAKE2b-256 | 87b96141303d6c4ed091912e7870c738be0cc998bc43c9bdd97b6fc0c6428926 |
File details
Details for the file fitsdataset-0.0.1-py3-none-any.whl.
File metadata
- Download URL: fitsdataset-0.0.1-py3-none-any.whl
- Upload date:
- Size: 4.9 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.2.0 pkginfo/1.5.0.1 requests/2.24.0 setuptools/50.3.0 requests-toolbelt/0.9.1 tqdm/4.48.2 CPython/3.8.1
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 92d48f71339ef87939048bebcafbc27f6865b31fc1130a98af7bf71b73c598b2 |
|
| MD5 | 142b3889007a6b57ff2338634d4ef2d0 |
|
| BLAKE2b-256 | d27d0522ab6a6a40dd7b6f2e67c94059a4e856abac7503d86dc7391350912355 |
