covid-19-repo-data icon indicating copy to clipboard operation
covid-19-repo-data copied to clipboard

PyTorch dataset for COVID19

Open AFAgarap opened this issue 3 years ago • 0 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.

AFAgarap avatar Apr 17 '21 09:04 AFAgarap