niftynet.contrib.dataset_sampler.image_window_dataset module

Creating tf.data.Dataset instance for image window sampler.

class ImageWindowDataset(reader=None, window_sizes=None, batch_size=1, windows_per_image=1, queue_length=10, shuffle=True, epoch=-1, name='image_dataset')

Bases: niftynet.layer.base_layer.Layer

This class creates a tf.data.Dataset instance from a sampler’s layer_op function or generator.

If from_generator, Dataset.from_generator interface will be used, Dataset.map interface will be used otherwise.

shapes

the sampler output (value of layer_op) is:

[windows_per_image, x, y, z, 1, channels]

returns a dictionary of sampler output shapes

tf_shapes

returns a dictionary of sampler output tensor shapes

tf_dtypes

returns a dictionary of sampler output tensorflow dtypes

layer_op(idx=None)

Generating each image as a window. Overriding this function to create new image sampling strategies.

This function should either yield a dictionary (for single window per image):

yield a dictionary

{
 'image_name': a numpy array,
 'image_name_location': (image_id,
                         x_start, y_start, z_start,
                         x_end, y_end, z_end)
}

or return a dictionary (for multiple windows per image):

return a dictionary:
{
 'image_name': a numpy array,
 'image_name_location': [n_samples, 7]
}

where the 7-element location vector encode the image_id, starting and ending coordinates of the image window.

Following the same notation, the dictionary can be extended to multiple modalities; the keys will be:

{'image_name_1', 'image_name_location_1',
 'image_name_2', 'image_name_location_2', ...}

Parameters:idx – image_id used to load the image at the i-th row of the input Returns:a image data dictionary

run_threads(session=None, *_unused_args, **_unused_argvs)

To be called at the beginning of running graph variables, to initialise dataset iterator.

pop_batch_op()

This function is used when connecting a sampler output to a network. e.g.:

data_dict = self.get_sampler()[0].pop_batch_op(device_id)
net_output = net_model(data_dict['image'], is_training)

Caution

Note it squeezes the output tensor of 6 dims [batch, x, y, z, time, modality] by removing all dims along which length is one.

Returns:a dictionary of image window tensors.

init_dataset()

Make a window samples dataset from the reader and layer_op. This function sets self.dataset.

Returns:

dataset_preprocessing(dataset)

dataset: batch and shuffle

Parameters:dataset – a tf.data.Dataset instance Returns:a tf.data.Dataset instance

close_all()

For compatibility with the queue-based sampler.

classmethod dummy_coordinates(image_id, image_sizes, n_samples)

This function returns a set of image window coordinates which are just from 0 to image_shapes.

Returns:a numpy array of n_samples spatial coordinates