def _plot_saliency_for_2d3d_radar(
list_of_input_matrices, list_of_saliency_matrices,
training_option_dict, saliency_colour_map_object,
max_colour_value_by_example, output_dir_name, storm_ids=None,
storm_times_unix_sec=None):
"""Plots saliency for 2-D azimuthal-shear and 3-D reflectivity fields.
E = number of examples (storm objects)
If `storm_ids is None` and `storm_times_unix_sec is None`, will assume that
the input matrices contain probability-matched means.
:param list_of_input_matrices: See doc for
`saliency_maps.read_standard_file`.
:param list_of_saliency_matrices: Same.
:param training_option_dict: Dictionary returned by
`cnn.read_model_metadata`.
:param saliency_colour_map_object: See documentation at top of file.
:param max_colour_value_by_example: length-E numpy array with max value in
colour scheme for each example. Minimum value for [i]th example will be
-1 * max_colour_value_by_example[i], since the colour scheme is
zero-centered and divergent.
:param output_dir_name: Name of output directory (figures will be saved
here).
:param storm_ids: length-E list of storm IDs (strings).
:param storm_times_unix_sec: length-E numpy array of storm times.
"""
pmm_flag = storm_ids is None and storm_times_unix_sec is None
reflectivity_matrix_dbz = list_of_input_matrices[0]
reflectivity_saliency_matrix = list_of_saliency_matrices[0]
az_shear_matrix_s01 = list_of_input_matrices[1]
az_shear_saliency_matrix = list_of_saliency_matrices[1]
num_examples = reflectivity_matrix_dbz.shape[0]
num_reflectivity_heights = len(
training_option_dict[trainval_io.RADAR_HEIGHTS_KEY]
)
num_panel_rows_for_reflectivity = int(numpy.floor(
numpy.sqrt(num_reflectivity_heights)
))
az_shear_field_names = training_option_dict[trainval_io.RADAR_FIELDS_KEY]
num_az_shear_fields = len(az_shear_field_names)
plot_colour_bar_flags = numpy.full(num_az_shear_fields, False, dtype=bool)
for i in range(num_examples):
_, these_axes_objects = radar_plotting.plot_3d_grid_without_coords(
field_matrix=numpy.flip(reflectivity_matrix_dbz[i, ..., 0], axis=0),
field_name=radar_utils.REFL_NAME,
grid_point_heights_metres=training_option_dict[
trainval_io.RADAR_HEIGHTS_KEY],
ground_relative=True,
num_panel_rows=num_panel_rows_for_reflectivity,
font_size=FONT_SIZE_SANS_COLOUR_BARS)
saliency_plotting.plot_many_2d_grids_with_pm_signs(
saliency_matrix_3d=numpy.flip(
reflectivity_saliency_matrix[i, ..., 0], axis=0),
axes_objects_2d_list=these_axes_objects,
colour_map_object=saliency_colour_map_object,
max_absolute_colour_value=max_colour_value_by_example[i])
this_colour_map_object, this_colour_norm_object = (
radar_plotting.get_default_colour_scheme(radar_utils.REFL_NAME)
)
plotting_utils.add_colour_bar(
axes_object_or_list=these_axes_objects,
values_to_colour=reflectivity_matrix_dbz[i, ..., 0],
colour_map=this_colour_map_object,
colour_norm_object=this_colour_norm_object,
orientation='horizontal', extend_min=True, extend_max=True)
if pmm_flag:
this_title_string = 'Probability-matched mean'
this_file_name = '{0:s}/saliency_pmm_reflectivity.jpg'.format(
output_dir_name)
else:
this_storm_time_string = time_conversion.unix_sec_to_string(
storm_times_unix_sec[i], TIME_FORMAT)
this_title_string = 'Storm "{0:s}" at {1:s}'.format(
storm_ids[i], this_storm_time_string)
this_file_name = (
'{0:s}/saliency_{1:s}_{2:s}_reflectivity.jpg'
).format(
output_dir_name, storm_ids[i].replace('_', '-'),
this_storm_time_string
)
this_title_string += ' (max absolute saliency = {0:.3f})'.format(
max_colour_value_by_example[i])
pyplot.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
print 'Saving figure to file: "{0:s}"...'.format(this_file_name)
pyplot.savefig(this_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
_, these_axes_objects = (
radar_plotting.plot_many_2d_grids_without_coords(
field_matrix=numpy.flip(az_shear_matrix_s01[i, ...], axis=0),
field_name_by_panel=az_shear_field_names,
panel_names=az_shear_field_names, num_panel_rows=1,
plot_colour_bar_by_panel=plot_colour_bar_flags,
font_size=FONT_SIZE_SANS_COLOUR_BARS)
)
saliency_plotting.plot_many_2d_grids_with_pm_signs(
saliency_matrix_3d=numpy.flip(
az_shear_saliency_matrix[i, ...], axis=0),
axes_objects_2d_list=these_axes_objects,
colour_map_object=saliency_colour_map_object,
max_absolute_colour_value=max_colour_value_by_example[i])
this_colour_map_object, this_colour_norm_object = (
radar_plotting.get_default_colour_scheme(
radar_utils.LOW_LEVEL_SHEAR_NAME)
)
plotting_utils.add_colour_bar(
axes_object_or_list=these_axes_objects,
values_to_colour=az_shear_saliency_matrix[i, ...],
colour_map=this_colour_map_object,
colour_norm_object=this_colour_norm_object,
orientation='horizontal', extend_min=True, extend_max=True)
pyplot.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
this_file_name = this_file_name.replace(
'_reflectivity.jpg', '_azimuthal-shear.jpg')
print 'Saving figure to file: "{0:s}"...'.format(this_file_name)
pyplot.savefig(this_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
def _plot_2d_radar(model_metadata_dict,
output_dir_name,
pmm_flag,
diff_colour_map_object=None,
max_colour_percentile_for_diff=None,
full_id_strings=None,
storm_time_strings=None,
novel_radar_matrix=None,
novel_radar_matrix_upconv=None,
novel_radar_matrix_upconv_svd=None):
"""Plots results of novelty detection for 2-D radar fields.
E = number of examples (storm objects)
M = number of rows in spatial grid
N = number of columns in spatial grid
C = number of channels (field/height pairs)
This method handles the 3 input matrices in the same way as
`_plot_3d_radar`.
:param model_metadata_dict: Dictionary returned by
`cnn.read_model_metadata`.
:param output_dir_name: Same.
:param pmm_flag: Same.
:param diff_colour_map_object: Same.
:param max_colour_percentile_for_diff: Same.
:param full_id_strings: Same.
:param storm_time_strings: Same.
:param novel_radar_matrix: E-by-M-by-N-by-C numpy array of original
(not reconstructed) radar fields.
:param novel_radar_matrix_upconv: E-by-M-by-N-by-C numpy array of
upconvnet-reconstructed radar fields.
:param novel_radar_matrix_upconv_svd: E-by-M-by-N-by-C numpy array of
upconvnet-and-SVD-reconstructed radar fields.
"""
training_option_dict = model_metadata_dict[cnn.TRAINING_OPTION_DICT_KEY]
list_of_layer_operation_dicts = model_metadata_dict[
cnn.LAYER_OPERATIONS_KEY]
if pmm_flag:
have_storm_ids = False
else:
have_storm_ids = not (full_id_strings is None
or storm_time_strings is None)
plot_difference = False
if novel_radar_matrix is not None:
plot_type_abbrev = 'actual'
plot_type_verbose = 'actual'
radar_matrix_to_plot = novel_radar_matrix
else:
if (novel_radar_matrix_upconv is not None
and novel_radar_matrix_upconv_svd is not None):
plot_difference = True
plot_type_abbrev = 'novelty'
plot_type_verbose = 'novelty'
radar_matrix_to_plot = (novel_radar_matrix_upconv -
novel_radar_matrix_upconv_svd)
else:
if novel_radar_matrix_upconv is not None:
plot_type_abbrev = 'upconv'
plot_type_verbose = 'upconvnet reconstruction'
radar_matrix_to_plot = novel_radar_matrix_upconv
else:
plot_type_abbrev = 'upconv-svd'
plot_type_verbose = 'upconvnet/SVD reconstruction'
radar_matrix_to_plot = novel_radar_matrix_upconv_svd
if list_of_layer_operation_dicts is None:
field_name_by_panel = training_option_dict[
trainval_io.RADAR_FIELDS_KEY]
panel_names = (radar_plotting.radar_fields_and_heights_to_panel_names(
field_names=field_name_by_panel,
heights_m_agl=training_option_dict[trainval_io.RADAR_HEIGHTS_KEY]))
plot_colour_bar_by_panel = numpy.full(len(panel_names),
True,
dtype=bool)
else:
field_name_by_panel, panel_names = (
radar_plotting.layer_ops_to_field_and_panel_names(
list_of_layer_operation_dicts))
plot_colour_bar_by_panel = numpy.full(len(panel_names),
False,
dtype=bool)
plot_colour_bar_by_panel[2::3] = True
num_panels = len(panel_names)
num_storms = radar_matrix_to_plot.shape[0]
num_channels = radar_matrix_to_plot.shape[-1]
num_panel_rows = int(numpy.floor(numpy.sqrt(num_channels)))
for i in range(num_storms):
if pmm_flag:
this_title_string = 'Probability-matched mean'
this_file_name = 'pmm'
else:
if have_storm_ids:
this_title_string = 'Storm "{0:s}" at {1:s}'.format(
full_id_strings[i], storm_time_strings[i])
this_file_name = '{0:s}_{1:s}'.format(
full_id_strings[i].replace('_', '-'),
storm_time_strings[i])
else:
this_title_string = 'Example {0:d}'.format(i + 1)
this_file_name = 'example{0:06d}'.format(i)
this_title_string += ' ({0:s})'.format(plot_type_verbose)
this_file_name = '{0:s}/{1:s}_{2:s}_radar.jpg'.format(
output_dir_name, this_file_name, plot_type_abbrev)
if plot_difference:
this_cmap_object_by_panel = [diff_colour_map_object] * num_panels
this_cnorm_object_by_panel = [None] * num_panels
if list_of_layer_operation_dicts is None:
for j in range(num_panels):
this_max_value = numpy.percentile(
numpy.absolute(radar_matrix_to_plot[i, ..., j]),
max_colour_percentile_for_diff)
this_cnorm_object_by_panel[
j] = matplotlib.colors.Normalize(vmin=-1 *
this_max_value,
vmax=this_max_value,
clip=False)
else:
unique_field_names = numpy.unique(
numpy.array(field_name_by_panel))
for this_field_name in unique_field_names:
these_panel_indices = numpy.where(
numpy.array(field_name_by_panel) == this_field_name)[0]
this_diff_matrix = radar_matrix_to_plot[
i, ..., these_panel_indices]
this_max_value = numpy.percentile(
numpy.absolute(this_diff_matrix),
max_colour_percentile_for_diff)
for this_index in these_panel_indices:
this_cnorm_object_by_panel[this_index] = (
matplotlib.colors.Normalize(vmin=-1 *
this_max_value,
vmax=this_max_value,
clip=False))
else:
this_cmap_object_by_panel = None
this_cnorm_object_by_panel = None
radar_plotting.plot_many_2d_grids_without_coords(
field_matrix=numpy.flip(radar_matrix_to_plot[i, ...], axis=0),
field_name_by_panel=field_name_by_panel,
num_panel_rows=num_panel_rows,
panel_names=panel_names,
colour_map_object_by_panel=this_cmap_object_by_panel,
colour_norm_object_by_panel=this_cnorm_object_by_panel,
plot_colour_bar_by_panel=plot_colour_bar_by_panel,
font_size=FONT_SIZE_WITH_COLOUR_BARS,
row_major=False)
pyplot.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
print('Saving figure to: "{0:s}"...'.format(this_file_name))
pyplot.savefig(this_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
def _plot_comparison(
predictor_matrix, model_metadata_dict, machine_mask_matrix_3d,
human_mask_matrix_3d, iou_by_channel, positive_flag, output_file_name):
"""Plots comparison between human and machine interpretation maps.
M = number of rows in grid (physical space)
N = number of columns in grid (physical space)
C = number of channels
:param predictor_matrix: M-by-N-by-C numpy array of predictors.
:param model_metadata_dict: Dictionary returned by
`cnn.read_model_metadata`.
:param machine_mask_matrix_3d: M-by-N-by-C numpy array of Boolean flags,
indicating where machine interpretation value is strongly positive or
negative.
:param human_mask_matrix_3d: Same.
:param iou_by_channel: length-C numpy array of IoU values (intersection over
union) between human and machine masks.
:param positive_flag: Boolean flag. If True (False), masks indicate where
interpretation value is strongly positive (negative).
:param output_file_name: Path to output file (figure will be saved here).
"""
training_option_dict = model_metadata_dict[cnn.TRAINING_OPTION_DICT_KEY]
list_of_layer_operation_dicts = model_metadata_dict[
cnn.LAYER_OPERATIONS_KEY]
if list_of_layer_operation_dicts is None:
field_name_by_panel = training_option_dict[trainval_io.RADAR_FIELDS_KEY]
panel_names = radar_plotting.fields_and_heights_to_names(
field_names=field_name_by_panel,
heights_m_agl=training_option_dict[trainval_io.RADAR_HEIGHTS_KEY]
)
plot_colour_bar_by_panel = numpy.full(
len(field_name_by_panel), True, dtype=bool
)
else:
field_name_by_panel, panel_names = (
radar_plotting.layer_operations_to_names(
list_of_layer_operation_dicts=list_of_layer_operation_dicts
)
)
plot_colour_bar_by_panel = numpy.full(
len(field_name_by_panel), False, dtype=bool
)
plot_colour_bar_by_panel[2::3] = True
num_panels = len(field_name_by_panel)
num_panel_rows = int(numpy.floor(
numpy.sqrt(num_panels)
))
for k in range(num_panels):
panel_names[k] += '\n{0:s} IoU = {1:.3f}'.format(
'Positive' if positive_flag else 'Negative',
iou_by_channel[k]
)
axes_object_matrix = radar_plotting.plot_many_2d_grids_without_coords(
field_matrix=numpy.flip(predictor_matrix, axis=0),
field_name_by_panel=field_name_by_panel, panel_names=panel_names,
num_panel_rows=num_panel_rows,
plot_colour_bar_by_panel=plot_colour_bar_by_panel, font_size=14,
row_major=False
)[1]
for k in range(num_panels):
_plot_comparison_one_channel(
human_mask_matrix_3d=human_mask_matrix_3d,
machine_mask_matrix_3d=numpy.flip(machine_mask_matrix_3d, axis=0),
channel_index=k, axes_object_matrix=axes_object_matrix)
print('Saving figure to: "{0:s}"...'.format(output_file_name))
pyplot.savefig(output_file_name, dpi=FIGURE_RESOLUTION_DPI,
pad_inches=0., bbox_inches='tight')
pyplot.close()
def _plot_saliency_for_2d_radar(
radar_matrix, radar_saliency_matrix, model_metadata_dict,
saliency_colour_map_object, max_colour_value_by_example,
output_dir_name, storm_ids=None, storm_times_unix_sec=None):
"""Plots saliency for 2-D radar fields.
E = number of examples
M = number of rows in spatial grid
N = number of columns in spatial grid
C = number of channels (field/height pairs)
If `storm_ids is None` and `storm_times_unix_sec is None`, will assume that
the input matrices contain probability-matched means.
:param radar_matrix: E-by-M-by-N-by-C numpy array of radar values
(predictors).
:param radar_saliency_matrix: E-by-M-by-N-by-C numpy array of saliency
values.
:param model_metadata_dict: See doc for `cnn.read_model_metadata`.
:param saliency_colour_map_object: See doc for
`_plot_saliency_for_2d3d_radar`.
:param max_colour_value_by_example: Same.
:param output_dir_name: Same.
:param storm_ids: Same.
:param storm_times_unix_sec: Same.
"""
pmm_flag = storm_ids is None and storm_times_unix_sec is None
training_option_dict = model_metadata_dict[cnn.TRAINING_OPTION_DICT_KEY]
list_of_layer_operation_dicts = model_metadata_dict[
cnn.LAYER_OPERATIONS_KEY]
if list_of_layer_operation_dicts is None:
field_name_by_panel = training_option_dict[
trainval_io.RADAR_FIELDS_KEY]
panel_names = (
radar_plotting.radar_fields_and_heights_to_panel_names(
field_names=field_name_by_panel,
heights_m_agl=training_option_dict[
trainval_io.RADAR_HEIGHTS_KEY]
)
)
plot_colour_bar_by_panel = numpy.full(
len(panel_names), True, dtype=bool)
else:
field_name_by_panel, panel_names = (
radar_plotting.layer_ops_to_field_and_panel_names(
list_of_layer_operation_dicts)
)
plot_colour_bar_by_panel = numpy.full(
len(panel_names), False, dtype=bool)
plot_colour_bar_by_panel[2::3] = True
num_examples = radar_matrix.shape[0]
num_panels = len(field_name_by_panel)
num_panel_rows = int(numpy.floor(numpy.sqrt(num_panels)))
for i in range(num_examples):
_, these_axes_objects = (
radar_plotting.plot_many_2d_grids_without_coords(
field_matrix=numpy.flip(radar_matrix[i, ...], axis=0),
field_name_by_panel=field_name_by_panel,
panel_names=panel_names, num_panel_rows=num_panel_rows,
plot_colour_bar_by_panel=plot_colour_bar_by_panel,
font_size=FONT_SIZE_WITH_COLOUR_BARS, row_major=False)
)
this_contour_interval = (
max_colour_value_by_example[i] / HALF_NUM_CONTOURS
)
saliency_plotting.plot_many_2d_grids_with_contours(
saliency_matrix_3d=numpy.flip(
radar_saliency_matrix[i, ...], axis=0),
axes_objects_2d_list=these_axes_objects,
colour_map_object=saliency_colour_map_object,
max_absolute_contour_level=max_colour_value_by_example[i],
contour_interval=this_contour_interval, row_major=False)
if pmm_flag:
this_title_string = 'Probability-matched mean'
this_file_name = '{0:s}/saliency_pmm_radar.jpg'.format(
output_dir_name)
else:
this_storm_time_string = time_conversion.unix_sec_to_string(
storm_times_unix_sec[i], TIME_FORMAT)
this_title_string = 'Storm "{0:s}" at {1:s}'.format(
storm_ids[i], this_storm_time_string)
this_file_name = '{0:s}/saliency_{1:s}_{2:s}_radar.jpg'.format(
output_dir_name, storm_ids[i].replace('_', '-'),
this_storm_time_string)
this_title_string += ' (max absolute saliency = {0:.3f})'.format(
max_colour_value_by_example[i])
pyplot.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
print 'Saving figure to file: "{0:s}"...'.format(this_file_name)
pyplot.savefig(this_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
def _plot_bwo_for_2d3d_radar(
list_of_optimized_matrices, training_option_dict,
diff_colour_map_object, max_colour_percentile_for_diff,
top_output_dir_name, pmm_flag, list_of_input_matrices=None,
storm_ids=None, storm_times_unix_sec=None):
"""Plots BWO results for 2-D azimuthal-shear and 3-D reflectivity fields.
E = number of examples (storm objects)
T = number of input tensors to the model
:param list_of_optimized_matrices: length-T list of numpy arrays, where the
[i]th array is the optimized version of the [i]th input matrix to the
model.
:param training_option_dict: See doc for `cnn.read_model_metadata`.
:param diff_colour_map_object: See documentation at top of file.
:param max_colour_percentile_for_diff: Same.
:param top_output_dir_name: Path to top-level output directory (figures will
be saved here).
:param pmm_flag: Boolean flag. If True, `list_of_predictor_matrices`
contains probability-matched means.
:param list_of_input_matrices: Same as `list_of_optimized_matrices` but with
non-optimized input matrices.
:param storm_ids: [optional and used only if `pmm_flag = False`]
length-E list of storm IDs (strings).
:param storm_times_unix_sec: [optional and used only if `pmm_flag = False`]
length-E numpy array of storm times.
"""
before_optimization_dir_name = '{0:s}/before_optimization'.format(
top_output_dir_name)
after_optimization_dir_name = '{0:s}/after_optimization'.format(
top_output_dir_name)
difference_dir_name = '{0:s}/after_minus_before_optimization'.format(
top_output_dir_name)
file_system_utils.mkdir_recursive_if_necessary(
directory_name=before_optimization_dir_name)
file_system_utils.mkdir_recursive_if_necessary(
directory_name=after_optimization_dir_name)
file_system_utils.mkdir_recursive_if_necessary(
directory_name=difference_dir_name)
if pmm_flag:
have_storm_ids = False
else:
have_storm_ids = not (storm_ids is None or storm_times_unix_sec is None)
az_shear_field_names = training_option_dict[trainval_io.RADAR_FIELDS_KEY]
num_az_shear_fields = len(az_shear_field_names)
plot_colour_bar_flags = numpy.full(num_az_shear_fields, False, dtype=bool)
num_storms = list_of_optimized_matrices[0].shape[0]
for i in range(num_storms):
print '\n'
if pmm_flag:
this_base_title_string = 'Probability-matched mean'
this_base_pathless_file_name = 'pmm'
else:
if have_storm_ids:
this_storm_time_string = time_conversion.unix_sec_to_string(
storm_times_unix_sec[i], TIME_FORMAT)
this_base_title_string = 'Storm "{0:s}" at {1:s}'.format(
storm_ids[i], this_storm_time_string)
this_base_pathless_file_name = '{0:s}_{1:s}'.format(
storm_ids[i].replace('_', '-'), this_storm_time_string)
else:
this_base_title_string = 'Example {0:d}'.format(i + 1)
this_base_pathless_file_name = 'example{0:06d}'.format(i)
this_reflectivity_matrix_dbz = numpy.flip(
list_of_optimized_matrices[0][i, ..., 0], axis=0)
this_num_heights = this_reflectivity_matrix_dbz.shape[-1]
this_num_panel_rows = int(numpy.floor(
numpy.sqrt(this_num_heights)
))
_, these_axes_objects = radar_plotting.plot_3d_grid_without_coords(
field_matrix=this_reflectivity_matrix_dbz,
field_name=radar_utils.REFL_NAME,
grid_point_heights_metres=training_option_dict[
trainval_io.RADAR_HEIGHTS_KEY],
ground_relative=True, num_panel_rows=this_num_panel_rows,
font_size=FONT_SIZE_SANS_COLOUR_BARS)
this_colour_map_object, this_colour_norm_object = (
radar_plotting.get_default_colour_scheme(radar_utils.REFL_NAME)
)
plotting_utils.add_colour_bar(
axes_object_or_list=these_axes_objects,
values_to_colour=this_reflectivity_matrix_dbz,
colour_map=this_colour_map_object,
colour_norm_object=this_colour_norm_object,
orientation='horizontal', extend_min=True, extend_max=True)
this_title_string = '{0:s} (after optimization)'.format(
this_base_title_string)
this_file_name = (
'{0:s}/{1:s}_after-optimization_reflectivity.jpg'
).format(after_optimization_dir_name, this_base_pathless_file_name)
pyplot.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
print 'Saving figure to: "{0:s}"...'.format(this_file_name)
pyplot.savefig(this_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
this_az_shear_matrix_s01 = numpy.flip(
list_of_optimized_matrices[1][i, ..., 0], axis=0)
_, these_axes_objects = (
radar_plotting.plot_many_2d_grids_without_coords(
field_matrix=this_az_shear_matrix_s01,
field_name_by_panel=az_shear_field_names, num_panel_rows=1,
panel_names=az_shear_field_names,
plot_colour_bar_by_panel=plot_colour_bar_flags,
font_size=FONT_SIZE_SANS_COLOUR_BARS)
)
this_colour_map_object, this_colour_norm_object = (
radar_plotting.get_default_colour_scheme(
radar_utils.LOW_LEVEL_SHEAR_NAME)
)
plotting_utils.add_colour_bar(
axes_object_or_list=these_axes_objects,
values_to_colour=this_az_shear_matrix_s01,
colour_map=this_colour_map_object,
colour_norm_object=this_colour_norm_object,
orientation='horizontal', extend_min=True, extend_max=True)
this_title_string = '{0:s} (after optimization)'.format(
this_base_title_string)
this_file_name = (
'{0:s}/{1:s}_after-optimization_azimuthal-shear.jpg'
).format(after_optimization_dir_name, this_base_pathless_file_name)
pyplot.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
print 'Saving figure to: "{0:s}"...'.format(this_file_name)
pyplot.savefig(this_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
if list_of_input_matrices is None:
continue
this_reflectivity_matrix_dbz = numpy.flip(
list_of_input_matrices[0][i, ..., 0], axis=0)
_, these_axes_objects = radar_plotting.plot_3d_grid_without_coords(
field_matrix=this_reflectivity_matrix_dbz,
field_name=radar_utils.REFL_NAME,
grid_point_heights_metres=training_option_dict[
trainval_io.RADAR_HEIGHTS_KEY],
ground_relative=True, num_panel_rows=this_num_panel_rows,
font_size=FONT_SIZE_SANS_COLOUR_BARS)
this_colour_map_object, this_colour_norm_object = (
radar_plotting.get_default_colour_scheme(radar_utils.REFL_NAME)
)
plotting_utils.add_colour_bar(
axes_object_or_list=these_axes_objects,
values_to_colour=this_reflectivity_matrix_dbz,
colour_map=this_colour_map_object,
colour_norm_object=this_colour_norm_object,
orientation='horizontal', extend_min=True, extend_max=True)
this_title_string = '{0:s} (before optimization)'.format(
this_base_title_string)
this_file_name = (
'{0:s}/{1:s}_before-optimization_reflectivity.jpg'
).format(before_optimization_dir_name, this_base_pathless_file_name)
pyplot.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
print 'Saving figure to: "{0:s}"...'.format(this_file_name)
pyplot.savefig(this_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
this_az_shear_matrix_s01 = numpy.flip(
list_of_input_matrices[1][i, ..., 0], axis=0)
_, these_axes_objects = (
radar_plotting.plot_many_2d_grids_without_coords(
field_matrix=this_az_shear_matrix_s01,
field_name_by_panel=az_shear_field_names, num_panel_rows=1,
panel_names=az_shear_field_names,
plot_colour_bar_by_panel=plot_colour_bar_flags,
font_size=FONT_SIZE_SANS_COLOUR_BARS)
)
this_colour_map_object, this_colour_norm_object = (
radar_plotting.get_default_colour_scheme(
radar_utils.LOW_LEVEL_SHEAR_NAME)
)
plotting_utils.add_colour_bar(
axes_object_or_list=these_axes_objects,
values_to_colour=this_az_shear_matrix_s01,
colour_map=this_colour_map_object,
colour_norm_object=this_colour_norm_object,
orientation='horizontal', extend_min=True, extend_max=True)
this_title_string = '{0:s} (before optimization)'.format(
this_base_title_string)
this_file_name = (
'{0:s}/{1:s}_before-optimization_azimuthal-shear.jpg'
).format(before_optimization_dir_name, this_base_pathless_file_name)
pyplot.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
print 'Saving figure to: "{0:s}"...'.format(this_file_name)
pyplot.savefig(this_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
this_refl_diff_matrix_dbz = (
list_of_optimized_matrices[0][i, ..., 0] -
list_of_input_matrices[0][i, ..., 0]
)
this_refl_diff_matrix_dbz = numpy.flip(
this_refl_diff_matrix_dbz, axis=0)
this_max_value_dbz = numpy.percentile(
numpy.absolute(this_refl_diff_matrix_dbz),
max_colour_percentile_for_diff)
this_colour_norm_object = matplotlib.colors.Normalize(
vmin=-1 * this_max_value_dbz, vmax=this_max_value_dbz, clip=False)
_, these_axes_objects = radar_plotting.plot_3d_grid_without_coords(
field_matrix=this_refl_diff_matrix_dbz,
field_name=radar_utils.REFL_NAME,
grid_point_heights_metres=training_option_dict[
trainval_io.RADAR_HEIGHTS_KEY],
ground_relative=True, num_panel_rows=this_num_panel_rows,
font_size=FONT_SIZE_SANS_COLOUR_BARS,
colour_map_object=diff_colour_map_object,
colour_norm_object=this_colour_norm_object)
plotting_utils.add_colour_bar(
axes_object_or_list=these_axes_objects,
values_to_colour=this_refl_diff_matrix_dbz,
colour_map=diff_colour_map_object,
colour_norm_object=this_colour_norm_object,
orientation='horizontal', extend_min=True, extend_max=True)
this_title_string = '{0:s} (after minus before optimization)'.format(
this_base_title_string)
this_file_name = (
'{0:s}/{1:s}_optimization-diff_reflectivity.jpg'
).format(difference_dir_name, this_base_pathless_file_name)
pyplot.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
print 'Saving figure to: "{0:s}"...'.format(this_file_name)
pyplot.savefig(this_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
this_shear_diff_matrix_s01 = (
list_of_optimized_matrices[1][i, ..., 0] -
list_of_input_matrices[1][i, ..., 0]
)
this_shear_diff_matrix_s01 = numpy.flip(
this_shear_diff_matrix_s01, axis=0)
this_max_value_s01 = numpy.percentile(
numpy.absolute(this_shear_diff_matrix_s01),
max_colour_percentile_for_diff)
this_colour_norm_object = matplotlib.colors.Normalize(
vmin=-1 * this_max_value_s01, vmax=this_max_value_s01, clip=False)
_, these_axes_objects = (
radar_plotting.plot_many_2d_grids_without_coords(
field_matrix=this_shear_diff_matrix_s01,
field_name_by_panel=az_shear_field_names, num_panel_rows=1,
panel_names=az_shear_field_names,
colour_map_object_by_panel=
[diff_colour_map_object] * num_az_shear_fields,
colour_norm_object_by_panel=
[copy.deepcopy(this_colour_norm_object)] * num_az_shear_fields,
plot_colour_bar_by_panel=plot_colour_bar_flags,
font_size=FONT_SIZE_SANS_COLOUR_BARS)
)
plotting_utils.add_colour_bar(
axes_object_or_list=these_axes_objects,
values_to_colour=this_shear_diff_matrix_s01,
colour_map=diff_colour_map_object,
colour_norm_object=this_colour_norm_object,
orientation='horizontal', extend_min=True, extend_max=True)
this_title_string = '{0:s} (after minus before optimization)'.format(
this_base_title_string)
this_file_name = (
'{0:s}/{1:s}_optimization-diff_azimuthal-shear.jpg'
).format(difference_dir_name, this_base_pathless_file_name)
pyplot.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
print 'Saving figure to: "{0:s}"...'.format(this_file_name)
pyplot.savefig(this_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
def _plot_2d_radar_cams(
radar_matrix, model_metadata_dict, cam_colour_map_object,
max_colour_prctile_for_cam, output_dir_name,
class_activation_matrix=None, ggradcam_output_matrix=None,
storm_ids=None, storm_times_unix_sec=None):
"""Plots class-activation maps for 2-D radar data.
E = number of examples
M = number of rows in spatial grid
N = number of columns in spatial grid
C = number of channels (field/height pairs)
This method will plot either `class_activation_matrix` or
`ggradcam_output_matrix`, not both.
If `storm_ids is None` and `storm_times_unix_sec is None`, will assume that
the input matrices contain probability-matched means.
:param radar_matrix: E-by-M-by-N-by-C numpy array of radar values.
:param model_metadata_dict: See doc for `_plot_3d_radar_cams`.
:param cam_colour_map_object: Same.
:param max_colour_prctile_for_cam: Same.
:param output_dir_name: Same.
:param class_activation_matrix: E-by-M-by-N numpy array of class
activations.
:param ggradcam_output_matrix: E-by-M-by-N-by-C numpy array of output values
from guided Grad-CAM.
:param storm_ids: length-E list of storm IDs (strings).
:param storm_times_unix_sec: length-E numpy array of storm times.
"""
pmm_flag = storm_ids is None and storm_times_unix_sec is None
training_option_dict = model_metadata_dict[cnn.TRAINING_OPTION_DICT_KEY]
list_of_layer_operation_dicts = model_metadata_dict[
cnn.LAYER_OPERATIONS_KEY]
if list_of_layer_operation_dicts is None:
field_name_by_panel = training_option_dict[
trainval_io.RADAR_FIELDS_KEY]
panel_names = (
radar_plotting.radar_fields_and_heights_to_panel_names(
field_names=field_name_by_panel,
heights_m_agl=training_option_dict[
trainval_io.RADAR_HEIGHTS_KEY]
)
)
plot_colour_bar_by_panel = numpy.full(
len(panel_names), True, dtype=bool)
else:
field_name_by_panel, panel_names = (
radar_plotting.layer_ops_to_field_and_panel_names(
list_of_layer_operation_dicts)
)
plot_colour_bar_by_panel = numpy.full(
len(panel_names), False, dtype=bool)
plot_colour_bar_by_panel[2::3] = True
num_examples = radar_matrix.shape[0]
num_channels = radar_matrix.shape[-1]
num_panel_rows = int(numpy.floor(numpy.sqrt(num_channels)))
if class_activation_matrix is None:
quantity_string = 'max absolute guided Grad-CAM output'
pathless_file_name_prefix = 'guided-gradcam'
else:
quantity_string = 'max class activation'
pathless_file_name_prefix = 'gradcam'
for i in range(num_examples):
_, these_axes_objects = (
radar_plotting.plot_many_2d_grids_without_coords(
field_matrix=numpy.flip(radar_matrix[i, ...], axis=0),
field_name_by_panel=field_name_by_panel,
panel_names=panel_names, num_panel_rows=num_panel_rows,
plot_colour_bar_by_panel=plot_colour_bar_by_panel,
font_size=FONT_SIZE_WITH_COLOUR_BARS, row_major=False)
)
if class_activation_matrix is None:
this_matrix = ggradcam_output_matrix[i, ...]
this_max_contour_level = numpy.percentile(
numpy.absolute(this_matrix), max_colour_prctile_for_cam)
if this_max_contour_level == 0:
this_max_contour_level = 10.
saliency_plotting.plot_many_2d_grids_with_contours(
saliency_matrix_3d=numpy.flip(this_matrix, axis=0),
axes_objects_2d_list=these_axes_objects,
colour_map_object=cam_colour_map_object,
max_absolute_contour_level=this_max_contour_level,
contour_interval=this_max_contour_level / 10, row_major=False)
else:
this_matrix = numpy.expand_dims(
class_activation_matrix[i, ...], axis=-1)
this_matrix = numpy.repeat(
this_matrix, repeats=num_channels, axis=-1)
this_max_contour_level = numpy.percentile(
this_matrix, max_colour_prctile_for_cam)
if this_max_contour_level == 0:
this_max_contour_level = 10.
cam_plotting.plot_many_2d_grids(
class_activation_matrix_3d=numpy.flip(this_matrix, axis=0),
axes_objects_2d_list=these_axes_objects,
colour_map_object=cam_colour_map_object,
max_contour_level=this_max_contour_level,
contour_interval=this_max_contour_level / NUM_CONTOURS,
row_major=False)
if pmm_flag:
this_title_string = 'Probability-matched mean'
this_figure_file_name = '{0:s}/{1:s}_pmm_radar.jpg'.format(
output_dir_name, pathless_file_name_prefix)
else:
this_storm_time_string = time_conversion.unix_sec_to_string(
storm_times_unix_sec[i], TIME_FORMAT)
this_title_string = 'Storm "{0:s}" at {1:s}'.format(
storm_ids[i], this_storm_time_string)
this_figure_file_name = '{0:s}/{1:s}_{2:s}_{3:s}_radar.jpg'.format(
output_dir_name, pathless_file_name_prefix,
storm_ids[i].replace('_', '-'), this_storm_time_string)
this_title_string += ' ({0:s} = {1:.3f})'.format(
quantity_string, this_max_contour_level)
pyplot.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
print 'Saving figure to file: "{0:s}"...'.format(this_figure_file_name)
pyplot.savefig(this_figure_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
def _plot_bwo_for_2d_radar(
optimized_radar_matrix, model_metadata_dict, diff_colour_map_object,
max_colour_percentile_for_diff, top_output_dir_name, pmm_flag,
input_radar_matrix=None, storm_ids=None, storm_times_unix_sec=None):
"""Plots BWO results for 2-D radar fields.
E = number of examples (storm objects)
M = number of rows in spatial grid
N = number of columns in spatial grid
C = number of channels (field/height pairs)
:param optimized_radar_matrix: E-by-M-by-N-by-C numpy array of radar values
(predictors).
:param model_metadata_dict: Dictionary returned by
`cnn.read_model_metadata`.
:param diff_colour_map_object: See doc for `_plot_bwo_for_2d3d_radar`.
:param max_colour_percentile_for_diff: Same.
:param top_output_dir_name: Same.
:param pmm_flag: Same.
:param input_radar_matrix: Same as `optimized_radar_matrix` but with
non-optimized input.
:param storm_ids: See doc for `_plot_bwo_for_2d3d_radar`.
:param storm_times_unix_sec: Same.
"""
before_optimization_dir_name = '{0:s}/before_optimization'.format(
top_output_dir_name)
after_optimization_dir_name = '{0:s}/after_optimization'.format(
top_output_dir_name)
difference_dir_name = '{0:s}/after_minus_before_optimization'.format(
top_output_dir_name)
file_system_utils.mkdir_recursive_if_necessary(
directory_name=before_optimization_dir_name)
file_system_utils.mkdir_recursive_if_necessary(
directory_name=after_optimization_dir_name)
file_system_utils.mkdir_recursive_if_necessary(
directory_name=difference_dir_name)
training_option_dict = model_metadata_dict[cnn.TRAINING_OPTION_DICT_KEY]
list_of_layer_operation_dicts = model_metadata_dict[
cnn.LAYER_OPERATIONS_KEY]
if pmm_flag:
have_storm_ids = False
else:
have_storm_ids = not (storm_ids is None or storm_times_unix_sec is None)
if list_of_layer_operation_dicts is None:
field_name_by_panel = training_option_dict[
trainval_io.RADAR_FIELDS_KEY]
panel_names = (
radar_plotting.radar_fields_and_heights_to_panel_names(
field_names=field_name_by_panel,
heights_m_agl=training_option_dict[
trainval_io.RADAR_HEIGHTS_KEY]
)
)
plot_colour_bar_by_panel = numpy.full(
len(panel_names), True, dtype=bool)
else:
field_name_by_panel, panel_names = (
radar_plotting.layer_ops_to_field_and_panel_names(
list_of_layer_operation_dicts)
)
plot_colour_bar_by_panel = numpy.full(
len(panel_names), False, dtype=bool)
plot_colour_bar_by_panel[2::3] = True
num_panels = len(panel_names)
num_storms = optimized_radar_matrix.shape[0]
num_channels = optimized_radar_matrix.shape[-1]
num_panel_rows = int(numpy.floor(
numpy.sqrt(num_channels)
))
for i in range(num_storms):
print '\n'
if pmm_flag:
this_base_title_string = 'Probability-matched mean'
this_base_pathless_file_name = 'pmm'
else:
if have_storm_ids:
this_storm_time_string = time_conversion.unix_sec_to_string(
storm_times_unix_sec[i], TIME_FORMAT)
this_base_title_string = 'Storm "{0:s}" at {1:s}'.format(
storm_ids[i], this_storm_time_string)
this_base_pathless_file_name = '{0:s}_{1:s}'.format(
storm_ids[i].replace('_', '-'), this_storm_time_string)
else:
this_base_title_string = 'Example {0:d}'.format(i + 1)
this_base_pathless_file_name = 'example{0:06d}'.format(i)
radar_plotting.plot_many_2d_grids_without_coords(
field_matrix=numpy.flip(optimized_radar_matrix[i, ...], axis=0),
field_name_by_panel=field_name_by_panel,
num_panel_rows=num_panel_rows, panel_names=panel_names,
plot_colour_bar_by_panel=plot_colour_bar_by_panel,
font_size=FONT_SIZE_WITH_COLOUR_BARS, row_major=False)
this_title_string = '{0:s} (after optimization)'.format(
this_base_title_string)
this_file_name = '{0:s}/{1:s}_after-optimization_radar.jpg'.format(
after_optimization_dir_name, this_base_pathless_file_name)
pyplot.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
print 'Saving figure to: "{0:s}"...'.format(this_file_name)
pyplot.savefig(this_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
if input_radar_matrix is None:
continue
radar_plotting.plot_many_2d_grids_without_coords(
field_matrix=numpy.flip(input_radar_matrix[i, ...], axis=0),
field_name_by_panel=field_name_by_panel,
num_panel_rows=num_panel_rows, panel_names=panel_names,
plot_colour_bar_by_panel=plot_colour_bar_by_panel,
font_size=FONT_SIZE_WITH_COLOUR_BARS, row_major=False)
this_title_string = '{0:s} (before optimization)'.format(
this_base_title_string)
this_file_name = '{0:s}/{1:s}_before-optimization_radar.jpg'.format(
before_optimization_dir_name, this_base_pathless_file_name)
pyplot.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
print 'Saving figure to: "{0:s}"...'.format(this_file_name)
pyplot.savefig(this_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
this_cmap_object_by_panel = [diff_colour_map_object] * num_panels
this_cnorm_object_by_panel = [None] * num_panels
if list_of_layer_operation_dicts is None:
for j in range(num_panels):
this_diff_matrix = (
optimized_radar_matrix[i, ..., j] -
input_radar_matrix[i, ..., j]
)
this_max_value = numpy.percentile(
numpy.absolute(this_diff_matrix),
max_colour_percentile_for_diff)
this_cnorm_object_by_panel[j] = matplotlib.colors.Normalize(
vmin=-1 * this_max_value, vmax=this_max_value, clip=False)
else:
unique_field_names = numpy.unique(numpy.array(field_name_by_panel))
for this_field_name in unique_field_names:
these_panel_indices = numpy.where(
numpy.array(field_name_by_panel) == this_field_name
)[0]
this_diff_matrix = (
optimized_radar_matrix[i, ..., these_panel_indices] -
input_radar_matrix[i, ..., these_panel_indices]
)
this_max_value = numpy.percentile(
numpy.absolute(this_diff_matrix),
max_colour_percentile_for_diff)
for this_index in these_panel_indices:
this_cnorm_object_by_panel[this_index] = (
matplotlib.colors.Normalize(
vmin=-1 * this_max_value, vmax=this_max_value,
clip=False)
)
this_diff_matrix = (
optimized_radar_matrix[i, ...] - input_radar_matrix[i, ...]
)
radar_plotting.plot_many_2d_grids_without_coords(
field_matrix=numpy.flip(this_diff_matrix, axis=0),
field_name_by_panel=field_name_by_panel,
num_panel_rows=num_panel_rows, panel_names=panel_names,
colour_map_object_by_panel=this_cmap_object_by_panel,
colour_norm_object_by_panel=this_cnorm_object_by_panel,
plot_colour_bar_by_panel=plot_colour_bar_by_panel,
font_size=FONT_SIZE_WITH_COLOUR_BARS, row_major=False)
this_title_string = '{0:s} (after minus before optimization)'.format(
this_base_title_string)
this_file_name = '{0:s}/{1:s}_optimization-diff_radar.jpg'.format(
difference_dir_name, this_base_pathless_file_name)
pyplot.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
print 'Saving figure to: "{0:s}"...'.format(this_file_name)
pyplot.savefig(this_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
def _plot_2d_radar_scan(list_of_predictor_matrices,
model_metadata_dict,
allow_whitespace,
title_string=None):
"""Plots 2-D radar scan for one example.
J = number of panel rows in image
K = number of panel columns in image
:param list_of_predictor_matrices: See doc for `_plot_3d_radar_scan`.
:param model_metadata_dict: Same.
:param allow_whitespace: Same.
:param title_string: Same.
:return: figure_objects: length-1 list of figure handles (instances of
`matplotlib.figure.Figure`).
:return: axes_object_matrices: length-1 list. Each element is a J-by-K
numpy array of axes handles (instances of
`matplotlib.axes._subplots.AxesSubplot`).
"""
training_option_dict = model_metadata_dict[cnn.TRAINING_OPTION_DICT_KEY]
list_of_layer_operation_dicts = model_metadata_dict[
cnn.LAYER_OPERATIONS_KEY]
if list_of_layer_operation_dicts is None:
field_name_by_panel = training_option_dict[
trainval_io.RADAR_FIELDS_KEY]
num_panels = len(field_name_by_panel)
panel_names = radar_plotting.radar_fields_and_heights_to_panel_names(
field_names=field_name_by_panel,
heights_m_agl=training_option_dict[trainval_io.RADAR_HEIGHTS_KEY])
plot_cbar_by_panel = numpy.full(num_panels, True, dtype=bool)
else:
list_of_layer_operation_dicts = [
list_of_layer_operation_dicts[k] for k in LAYER_OP_INDICES_TO_KEEP
]
list_of_predictor_matrices[0] = list_of_predictor_matrices[0][
..., LAYER_OP_INDICES_TO_KEEP]
field_name_by_panel, panel_names = (
radar_plotting.layer_ops_to_field_and_panel_names(
list_of_layer_operation_dicts=list_of_layer_operation_dicts))
num_panels = len(field_name_by_panel)
plot_cbar_by_panel = numpy.full(num_panels, True, dtype=bool)
# if allow_whitespace:
# if len(field_name_by_panel) == 12:
# plot_cbar_by_panel[2::3] = True
# else:
# plot_cbar_by_panel[:] = True
num_panel_rows = int(numpy.floor(numpy.sqrt(num_panels)))
num_panel_columns = int(numpy.ceil(float(num_panels) / num_panel_rows))
if allow_whitespace:
figure_object = None
axes_object_matrix = None
else:
figure_object, axes_object_matrix = (
plotting_utils.create_paneled_figure(num_rows=num_panel_rows,
num_columns=num_panel_columns,
horizontal_spacing=0.,
vertical_spacing=0.,
shared_x_axis=False,
shared_y_axis=False,
keep_aspect_ratio=True))
figure_object, axes_object_matrix = (
radar_plotting.plot_many_2d_grids_without_coords(
field_matrix=numpy.flip(list_of_predictor_matrices[0], axis=0),
field_name_by_panel=field_name_by_panel,
panel_names=panel_names,
num_panel_rows=num_panel_rows,
figure_object=figure_object,
axes_object_matrix=axes_object_matrix,
plot_colour_bar_by_panel=plot_cbar_by_panel,
font_size=FONT_SIZE_WITH_COLOUR_BARS,
row_major=False))
if allow_whitespace and title_string is not None:
pyplot.suptitle(title_string, fontsize=TITLE_FONT_SIZE)
return [figure_object], [axes_object_matrix]
def _plot_2d3d_radar_scan(list_of_predictor_matrices,
model_metadata_dict,
allow_whitespace,
title_string=None):
"""Plots 3-D reflectivity and 2-D azimuthal shear for one example.
:param list_of_predictor_matrices: See doc for `_plot_3d_radar_scan`.
:param model_metadata_dict: Same.
:param allow_whitespace: Same.
:param title_string: Same.
:return: figure_objects: length-2 list of figure handles (instances of
`matplotlib.figure.Figure`). The first is for reflectivity; the second
is for azimuthal shear.
:return: axes_object_matrices: length-2 list (the first is for reflectivity;
the second is for azimuthal shear). Each element is a 2-D numpy
array of axes handles (instances of
`matplotlib.axes._subplots.AxesSubplot`).
"""
training_option_dict = model_metadata_dict[cnn.TRAINING_OPTION_DICT_KEY]
az_shear_field_names = training_option_dict[trainval_io.RADAR_FIELDS_KEY]
refl_heights_m_agl = training_option_dict[trainval_io.RADAR_HEIGHTS_KEY]
num_az_shear_fields = len(az_shear_field_names)
num_refl_heights = len(refl_heights_m_agl)
this_num_panel_rows = int(numpy.floor(numpy.sqrt(num_refl_heights)))
this_num_panel_columns = int(
numpy.ceil(float(num_refl_heights) / this_num_panel_rows))
if allow_whitespace:
refl_figure_object = None
refl_axes_object_matrix = None
else:
refl_figure_object, refl_axes_object_matrix = (
plotting_utils.create_paneled_figure(
num_rows=this_num_panel_rows,
num_columns=this_num_panel_columns,
horizontal_spacing=0.,
vertical_spacing=0.,
shared_x_axis=False,
shared_y_axis=False,
keep_aspect_ratio=True))
refl_figure_object, refl_axes_object_matrix = (
radar_plotting.plot_3d_grid_without_coords(
field_matrix=numpy.flip(list_of_predictor_matrices[0][..., 0],
axis=0),
field_name=radar_utils.REFL_NAME,
grid_point_heights_metres=refl_heights_m_agl,
ground_relative=True,
num_panel_rows=this_num_panel_rows,
figure_object=refl_figure_object,
axes_object_matrix=refl_axes_object_matrix,
font_size=FONT_SIZE_SANS_COLOUR_BARS))
if allow_whitespace:
this_colour_map_object, this_colour_norm_object = (
radar_plotting.get_default_colour_scheme(radar_utils.REFL_NAME))
plotting_utils.plot_colour_bar(
axes_object_or_matrix=refl_axes_object_matrix,
data_matrix=list_of_predictor_matrices[0],
colour_map_object=this_colour_map_object,
colour_norm_object=this_colour_norm_object,
orientation_string='horizontal',
extend_min=True,
extend_max=True)
if title_string is not None:
this_title_string = '{0:s}; {1:s}'.format(title_string,
radar_utils.REFL_NAME)
pyplot.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
if allow_whitespace:
shear_figure_object = None
shear_axes_object_matrix = None
else:
shear_figure_object, shear_axes_object_matrix = (
plotting_utils.create_paneled_figure(
num_rows=1,
num_columns=num_az_shear_fields,
horizontal_spacing=0.,
vertical_spacing=0.,
shared_x_axis=False,
shared_y_axis=False,
keep_aspect_ratio=True))
shear_figure_object, shear_axes_object_matrix = (
radar_plotting.plot_many_2d_grids_without_coords(
field_matrix=numpy.flip(list_of_predictor_matrices[1], axis=0),
field_name_by_panel=az_shear_field_names,
panel_names=az_shear_field_names,
num_panel_rows=1,
figure_object=shear_figure_object,
axes_object_matrix=shear_axes_object_matrix,
plot_colour_bar_by_panel=numpy.full(num_az_shear_fields,
False,
dtype=bool),
font_size=FONT_SIZE_SANS_COLOUR_BARS))
if allow_whitespace:
this_colour_map_object, this_colour_norm_object = (
radar_plotting.get_default_colour_scheme(
radar_utils.LOW_LEVEL_SHEAR_NAME))
plotting_utils.plot_colour_bar(
axes_object_or_matrix=shear_axes_object_matrix,
data_matrix=list_of_predictor_matrices[1],
colour_map_object=this_colour_map_object,
colour_norm_object=this_colour_norm_object,
orientation_string='horizontal',
extend_min=True,
extend_max=True)
if title_string is not None:
pyplot.suptitle(title_string, fontsize=TITLE_FONT_SIZE)
figure_objects = [refl_figure_object, shear_figure_object]
axes_object_matrices = [refl_axes_object_matrix, shear_axes_object_matrix]
return figure_objects, axes_object_matrices
def plot_examples(list_of_predictor_matrices,
storm_ids,
storm_times_unix_sec,
model_metadata_dict,
output_dir_name,
storm_activations=None):
"""Plots one or more learning examples.
E = number of examples (storm objects)
:param list_of_predictor_matrices: List created by
`testing_io.read_specific_examples`. Contains data to be plotted.
:param storm_ids: length-E list of storm IDs.
:param storm_times_unix_sec: length-E numpy array of storm times.
:param model_metadata_dict: See doc for `cnn.read_model_metadata`.
:param output_dir_name: Name of output directory (figures will be saved
here).
:param storm_activations: length-E numpy array of storm activations (may be
None). Will be included in title of each figure.
"""
training_option_dict = model_metadata_dict[cnn.TRAINING_OPTION_DICT_KEY]
sounding_field_names = training_option_dict[
trainval_io.SOUNDING_FIELDS_KEY]
plot_soundings = sounding_field_names is not None
if plot_soundings:
list_of_metpy_dictionaries = dl_utils.soundings_to_metpy_dictionaries(
sounding_matrix=list_of_predictor_matrices[-1],
field_names=sounding_field_names)
else:
list_of_metpy_dictionaries = None
num_radar_dimensions = len(list_of_predictor_matrices[0].shape) - 2
list_of_layer_operation_dicts = model_metadata_dict[
cnn.LAYER_OPERATIONS_KEY]
if num_radar_dimensions == 2:
if list_of_layer_operation_dicts is None:
field_name_by_panel = training_option_dict[
trainval_io.RADAR_FIELDS_KEY]
panel_names = (
radar_plotting.radar_fields_and_heights_to_panel_names(
field_names=field_name_by_panel,
heights_m_agl=training_option_dict[
trainval_io.RADAR_HEIGHTS_KEY]))
plot_colour_bar_by_panel = numpy.full(len(panel_names),
True,
dtype=bool)
else:
field_name_by_panel, panel_names = (
radar_plotting.layer_ops_to_field_and_panel_names(
list_of_layer_operation_dicts))
plot_colour_bar_by_panel = numpy.full(len(panel_names),
False,
dtype=bool)
plot_colour_bar_by_panel[2::3] = True
else:
field_name_by_panel = None
panel_names = None
plot_colour_bar_by_panel = None
az_shear_field_names = training_option_dict[trainval_io.RADAR_FIELDS_KEY]
num_az_shear_fields = len(az_shear_field_names)
num_storms = len(storm_ids)
myrorss_2d3d = len(list_of_predictor_matrices) == 3
for i in range(num_storms):
this_time_string = time_conversion.unix_sec_to_string(
storm_times_unix_sec[i], TIME_FORMAT)
this_base_title_string = 'Storm "{0:s}" at {1:s}'.format(
storm_ids[i], this_time_string)
if storm_activations is not None:
this_base_title_string += ' (activation = {0:.3f})'.format(
storm_activations[i])
this_base_file_name = '{0:s}/storm={1:s}_{2:s}'.format(
output_dir_name, storm_ids[i].replace('_', '-'), this_time_string)
if plot_soundings:
sounding_plotting.plot_sounding(
sounding_dict_for_metpy=list_of_metpy_dictionaries[i],
title_string=this_base_title_string)
this_file_name = '{0:s}_sounding.jpg'.format(this_base_file_name)
print 'Saving figure to: "{0:s}"...'.format(this_file_name)
pyplot.savefig(this_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
if myrorss_2d3d:
this_reflectivity_matrix_dbz = numpy.flip(
list_of_predictor_matrices[0][i, ..., 0], axis=0)
this_num_heights = this_reflectivity_matrix_dbz.shape[-1]
this_num_panel_rows = int(numpy.floor(
numpy.sqrt(this_num_heights)))
_, these_axes_objects = radar_plotting.plot_3d_grid_without_coords(
field_matrix=this_reflectivity_matrix_dbz,
field_name=radar_utils.REFL_NAME,
grid_point_heights_metres=training_option_dict[
trainval_io.RADAR_HEIGHTS_KEY],
ground_relative=True,
num_panel_rows=this_num_panel_rows,
font_size=FONT_SIZE_SANS_COLOUR_BARS)
this_colour_map_object, this_colour_norm_object = (
radar_plotting.get_default_colour_scheme(
radar_utils.REFL_NAME))
plotting_utils.add_colour_bar(
axes_object_or_list=these_axes_objects,
values_to_colour=this_reflectivity_matrix_dbz,
colour_map=this_colour_map_object,
colour_norm_object=this_colour_norm_object,
orientation='horizontal',
extend_min=True,
extend_max=True)
this_title_string = '{0:s}; {1:s}'.format(this_base_title_string,
radar_utils.REFL_NAME)
this_file_name = '{0:s}_reflectivity.jpg'.format(
this_base_file_name)
pyplot.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
print 'Saving figure to: "{0:s}"...'.format(this_file_name)
pyplot.savefig(this_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
this_az_shear_matrix_s01 = numpy.flip(
list_of_predictor_matrices[1][i, ..., 0], axis=0)
_, these_axes_objects = (
radar_plotting.plot_many_2d_grids_without_coords(
field_matrix=this_az_shear_matrix_s01,
field_name_by_panel=az_shear_field_names,
panel_names=az_shear_field_names,
num_panel_rows=1,
plot_colour_bar_by_panel=numpy.full(num_az_shear_fields,
False,
dtype=bool),
font_size=FONT_SIZE_SANS_COLOUR_BARS))
this_colour_map_object, this_colour_norm_object = (
radar_plotting.get_default_colour_scheme(
radar_utils.LOW_LEVEL_SHEAR_NAME))
plotting_utils.add_colour_bar(
axes_object_or_list=these_axes_objects,
values_to_colour=this_az_shear_matrix_s01,
colour_map=this_colour_map_object,
colour_norm_object=this_colour_norm_object,
orientation='horizontal',
extend_min=True,
extend_max=True)
this_file_name = '{0:s}_shear.jpg'.format(this_base_file_name)
pyplot.suptitle(this_base_title_string, fontsize=TITLE_FONT_SIZE)
print 'Saving figure to: "{0:s}"...'.format(this_file_name)
pyplot.savefig(this_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
continue
this_radar_matrix = list_of_predictor_matrices[0]
if num_radar_dimensions == 2:
this_num_channels = this_radar_matrix.shape[-1]
this_num_panel_rows = int(
numpy.floor(numpy.sqrt(this_num_channels)))
radar_plotting.plot_many_2d_grids_without_coords(
field_matrix=numpy.flip(this_radar_matrix[i, ...], axis=0),
field_name_by_panel=field_name_by_panel,
panel_names=panel_names,
num_panel_rows=this_num_panel_rows,
plot_colour_bar_by_panel=plot_colour_bar_by_panel,
font_size=FONT_SIZE_WITH_COLOUR_BARS,
row_major=False)
this_title_string = this_base_title_string + ''
pyplot.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
this_file_name = '{0:s}.jpg'.format(this_base_file_name)
print 'Saving figure to: "{0:s}"...'.format(this_file_name)
pyplot.savefig(this_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
continue
radar_field_names = training_option_dict[trainval_io.RADAR_FIELDS_KEY]
radar_heights_m_agl = training_option_dict[
trainval_io.RADAR_HEIGHTS_KEY]
for j in range(len(radar_field_names)):
this_num_heights = this_radar_matrix.shape[-2]
this_num_panel_rows = int(numpy.floor(
numpy.sqrt(this_num_heights)))
_, these_axes_objects = radar_plotting.plot_3d_grid_without_coords(
field_matrix=numpy.flip(this_radar_matrix[i, ..., j], axis=0),
field_name=radar_field_names[j],
grid_point_heights_metres=radar_heights_m_agl,
ground_relative=True,
num_panel_rows=this_num_panel_rows,
font_size=FONT_SIZE_SANS_COLOUR_BARS)
this_colour_map_object, this_colour_norm_object = (
radar_plotting.get_default_colour_scheme(radar_field_names[j]))
plotting_utils.add_colour_bar(
axes_object_or_list=these_axes_objects,
values_to_colour=this_radar_matrix[i, ..., j],
colour_map=this_colour_map_object,
colour_norm_object=this_colour_norm_object,
orientation='horizontal',
extend_min=True,
extend_max=True)
this_title_string = '{0:s}; {1:s}'.format(this_base_title_string,
radar_field_names[j])
this_file_name = '{0:s}_{1:s}.jpg'.format(
this_base_file_name, radar_field_names[j].replace('_', '-'))
pyplot.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
print 'Saving figure to: "{0:s}"...'.format(this_file_name)
pyplot.savefig(this_file_name, dpi=FIGURE_RESOLUTION_DPI)
pyplot.close()
def _plot_2d_radar_difference(difference_matrix,
colour_map_object,
max_colour_percentile,
model_metadata_dict,
backwards_opt_dict,
output_dir_name,
example_index=None,
significance_matrix=None):
"""Plots difference (after minus before optimization) for 2-D radar data.
M = number of rows in spatial grid
N = number of columns in spatial grid
C = number of channels
:param difference_matrix: M-by-N-by-C numpy array of differences (after
minus before optimization).
:param colour_map_object: See doc for `_plot_3d_radar_difference`.
:param max_colour_percentile: Same.
:param model_metadata_dict: Same.
:param backwards_opt_dict: Same.
:param output_dir_name: Same.
:param example_index: Same.
:param significance_matrix: M-by-N-by-C numpy array of Boolean flags,
indicating where these differences are significantly different than
differences from another backwards optimization.
"""
pmm_flag = backwards_opt.MEAN_FINAL_ACTIVATION_KEY in backwards_opt_dict
if pmm_flag:
initial_activation = backwards_opt_dict[
backwards_opt.MEAN_INITIAL_ACTIVATION_KEY]
final_activation = backwards_opt_dict[
backwards_opt.MEAN_FINAL_ACTIVATION_KEY]
full_storm_id_string = None
storm_time_string = None
else:
initial_activation = backwards_opt_dict[
backwards_opt.INITIAL_ACTIVATIONS_KEY][example_index]
final_activation = backwards_opt_dict[
backwards_opt.FINAL_ACTIVATIONS_KEY][example_index]
full_storm_id_string = backwards_opt_dict[
backwards_opt.FULL_IDS_KEY][example_index]
storm_time_string = time_conversion.unix_sec_to_string(
backwards_opt_dict[backwards_opt.STORM_TIMES_KEY][example_index],
plot_input_examples.TIME_FORMAT)
conv_2d3d = model_metadata_dict[cnn.CONV_2D3D_KEY]
training_option_dict = model_metadata_dict[cnn.TRAINING_OPTION_DICT_KEY]
if conv_2d3d:
num_fields = len(training_option_dict[trainval_io.RADAR_FIELDS_KEY])
radar_heights_m_agl = numpy.full(num_fields,
radar_utils.SHEAR_HEIGHT_M_ASL,
dtype=int)
else:
radar_heights_m_agl = training_option_dict[
trainval_io.RADAR_HEIGHTS_KEY]
list_of_layer_operation_dicts = model_metadata_dict[
cnn.LAYER_OPERATIONS_KEY]
if list_of_layer_operation_dicts is None:
field_name_by_panel = training_option_dict[
trainval_io.RADAR_FIELDS_KEY]
panel_names = radar_plotting.radar_fields_and_heights_to_panel_names(
field_names=field_name_by_panel, heights_m_agl=radar_heights_m_agl)
else:
field_name_by_panel, panel_names = (
radar_plotting.layer_ops_to_field_and_panel_names(
list_of_layer_operation_dicts=list_of_layer_operation_dicts))
num_panels = len(field_name_by_panel)
plot_cbar_by_panel = numpy.full(num_panels, True, dtype=bool)
cmap_object_by_panel = [colour_map_object] * num_panels
cnorm_object_by_panel = [None] * num_panels
for j in range(num_panels):
this_max_colour_value = numpy.percentile(
numpy.absolute(difference_matrix[..., j]), max_colour_percentile)
cnorm_object_by_panel[j] = matplotlib.colors.Normalize(
vmin=-1 * this_max_colour_value,
vmax=this_max_colour_value,
clip=False)
num_panel_rows = int(numpy.floor(numpy.sqrt(num_panels)))
figure_object, axes_object_matrix = (
radar_plotting.plot_many_2d_grids_without_coords(
field_matrix=numpy.flip(difference_matrix, axis=0),
field_name_by_panel=field_name_by_panel,
num_panel_rows=num_panel_rows,
panel_names=panel_names,
row_major=False,
colour_map_object_by_panel=cmap_object_by_panel,
colour_norm_object_by_panel=cnorm_object_by_panel,
plot_colour_bar_by_panel=plot_cbar_by_panel,
font_size=FONT_SIZE_WITH_COLOUR_BARS))
if significance_matrix is not None:
significance_plotting.plot_many_2d_grids_without_coords(
significance_matrix=numpy.flip(significance_matrix, axis=0),
axes_object_matrix=axes_object_matrix,
row_major=False)
if pmm_flag:
this_title_string = 'PMM'
else:
this_title_string = 'Storm "{0:s}" at {1:s}'.format(
full_storm_id_string, storm_time_string)
this_title_string += '; activation from {0:.2e} to {1:.2e}'.format(
initial_activation, final_activation)
figure_object.suptitle(this_title_string, fontsize=TITLE_FONT_SIZE)
output_file_name = plot_input_examples.metadata_to_radar_fig_file_name(
output_dir_name=output_dir_name,
pmm_flag=pmm_flag,
full_storm_id_string=full_storm_id_string,
storm_time_string=storm_time_string,
radar_field_name='shear' if conv_2d3d else None)
print('Saving figure to: "{0:s}"...'.format(output_file_name))
figure_object.savefig(output_file_name,
dpi=FIGURE_RESOLUTION_DPI,
pad_inches=0,
bbox_inches='tight')
pyplot.close(figure_object)