delta.data.predict_generator_seg#

delta.data.predict_generator_seg(files_path, *, files_list=None, target_size=(256, 32), crop_windows=False)[source]#

Get a generator for predicting segmentation on new image files once the segmentation U-Net has been trained.

Parameters
files_pathPath

Path to image files folder.

files_listlist/tuple of Paths, optional

List of file names to read in the folder. If None, all files in the folder will be read. The default is None.

target_sizetuple of 2 ints, optional

Size for the images to be resized. The default is (256,32).

crop_windowsbool

TODO The default is False.

Returns
mygengenerator

Generator that will yield single image files as 4D numpy arrays of size (1, target_size[0], target_size[1], 1).