File size: 945 Bytes
78a947a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
"""
Contains an implementation of a dataloader for preprocessed images.
Author: Ole-Christian Galbo Engstrøm
E-mail: ocge@foss.dk
"""
from torch.utils.data import DataLoader
from ..datasets.preprocessed_image_dataset import PreprocessedImageDataset
def get_preprocessed_image_dataloader(
reference_values,
images_path,
quant_biases_path,
quant_scales_path,
masks_path,
csv_path,
split,
transform,
batch_size,
shuffle,
):
dataset = PreprocessedImageDataset(
reference_values=reference_values,
images_path=images_path,
quant_biases_path=quant_biases_path,
quant_scales_path=quant_scales_path,
masks_path=masks_path,
csv_path=csv_path,
split=split,
split_column="split",
transform=transform,
)
dataloader = DataLoader(
dataset, batch_size=batch_size, shuffle=shuffle, num_workers=4
)
return dataloader
|