github / covid-19-repo-data

Data archive of identifiable COVID-19 related public projects on GitHub
Creative Commons Zero v1.0 Universal
509 stars 178 forks source link

PyTorch dataset for COVID19 #26

Open AFAgarap opened 3 years ago

AFAgarap commented 3 years ago

Please add my repository that allows users to load a COVID19 dataset (either binary or non-binary) and create a data loader for it in PyTorch.

from pt_datasets import load_dataset, create_dataloader

train_data, test_data = load_dataset(
    "binary_covid",
    preprocessed_covidx=True,
    preprocessing_bsize=256,
    image_size=64
)

# this is ready for use in a PyTorch model training
train_loader = create_dataloader(train_data, batch_size=256)  

The library can be installed via pip: pip install pt-datasets.