mldas.explore.prepare.set_creation

mldas.explore.prepare. set_creation ( datapath , img_size = 100 , stride = 10 , sample_size = 1 , adjust = False , shuffle = True , nrand = None , select = None , verbose = False )

Create PyTorch tensor training set of single-channel data images.

Parameters

datapath str

Path to data repository

img_size int

Size of squared image

stride int

Sliding interval

sample_size int

Number of input MAT data files to use

adjust bool

Do brightness adjustment

shuffle bool

Shuffling images in set

nrand int

Specify total number of images in final set

select int , list

Index(es) of specific image index

verbose bool

Do verbose

Returns

Xtrain torch.Tensor

Tensor of images