Skip to content
This repository has been archived by the owner on Sep 10, 2024. It is now read-only.

PyTorch dataset for COVID19 #26

Open
AFAgarap opened this issue Apr 17, 2021 · 0 comments
Open

PyTorch dataset for COVID19 #26

AFAgarap opened this issue Apr 17, 2021 · 0 comments

Comments

@AFAgarap
Copy link

AFAgarap commented Apr 17, 2021

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.

# for free to subscribe to this conversation on GitHub. Already have an account? #.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant