pyMSDtorch.viz_tools package

Submodules

pyMSDtorch.viz_tools.plots module

pyMSDtorch.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

pyMSDtorch.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

pyMSDtorch.viz_tools.plots.plot_shapes_data_numpy(shapes_dict)

This function takes the returned dict of numpy array image bundles from test_data/twoD/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

pyMSDtorch.viz_tools.plots.plot_training_results_regression(results_dict)
pyMSDtorch.viz_tools.plots.plot_training_results_segmentation(results_dict)

Module contents