niftynet.contrib.csv_reader.csv_reader module

class CSVReader(names=None)

Bases: niftynet.layer.base_layer.Layer

initialise(path_to_csv)

to_ohe(labels)

layer_op(idx=None, shuffle=True)

shapes

return – dict of label shape and label location shape

tf_dtypes

Infer input data dtypes in TF

tf_shapes

return – a dictionary of sampler output tensor shapes