github / covid-19-repo-data

Data archive of identifiable COVID-19 related public projects on GitHub

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

PyTorch dataset for COVID19

AFAgarap opened this issue · comments

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.