nondefaced_detector.dataloaders.dataset.get_dataset¶

get_dataset(file_pattern, n_classes, batch_size, volume_shape, plane, n_slices=24, block_shape=None, n_epochs=None, mapping=None, shuffle_buffer_size=None, num_parallel_calls=- 1, mode='train')[source]¶

Returns tf.data.Dataset after preprocessing from tfrecords for training and validation

Parameters