--- a +++ b/sandbox/test_iterators.py @@ -0,0 +1,15 @@ +from configuration import set_configuration, config +import utils_plots +import numpy as np + +set_configuration('configs_seg_scan', 'luna_s_local') + +data_iter = config().valid_data_iterator +for (x, y, lung_mask, annotations, transform_matrices, pid) in data_iter.generate(): + + predictions_scan = lung_mask * x + + for nodule_n, zyxd in enumerate(annotations): + utils_plots.plot_slice_3d_4(input=x[0, 0], lung_mask=lung_mask[0, 0], prediction=predictions_scan[0, 0], + mask=y[0, 0], + axis=0, pid='-'.join([str(nodule_n), str(pid)]), idx=zyxd)