dlsia.viz_tools package
Submodules
dlsia.viz_tools.plots module
- dlsia.viz_tools.plots.plot_heatmap_and_labels(img, labels, plot_params)
Make a side-by-side plot of an image and a set of labels
- Parameters:
img – The rgb image
labels – the heatmap with labels
plot_params – some mplot parameters in a dict
- Returns:
a graph object
- dlsia.viz_tools.plots.plot_rgb_and_labels(img, labels, plot_params)
Make a side-by-side plot of an RGB image and a set of labels
- Parameters:
img – The rgb image
labels – the heatmap with labels
plot_params – some mplot parameters in a dict
- Returns:
a graph object
- dlsia.viz_tools.plots.plot_shapes_data_numpy(shapes_dict)
This function takes the returned dict of numpy array image bundles from test_data/two_d/random_shapes/build_shape_set_numpy and plots the ground truth, noisy, and mask for each of the four shape classes
- Parameters:
shapes_dict – the shapes data dictionary containing ‘GroundTruth,’ ‘Noisy,’ ‘ClassImage,’ and ‘Labels’ numpy arrays
- dlsia.viz_tools.plots.plot_training_results_regression(results_dict)
- dlsia.viz_tools.plots.plot_training_results_segmentation(results_dict)