def save_attention_map(self, attention_map):
"""Saves an attention map."""
medcam_utils.save_attention_map(filename=self.medcam_dict['output_dir'] +
"/" + self.medcam_dict['current_layer'] +
"/attention_map_" +
str(self.medcam_dict['counter']),
attention_map=attention_map,
heatmap=self.medcam_dict['heatmap'])
self.medcam_dict['counter'] += 1
def save_attention_map(self, attention_map, raw_input=None):
"""Saves an attention map."""
medcam_utils.save_attention_map(filename=self.medcam_dict['output_dir'] +
"/" + self.medcam_dict['current_layer'] +
"/attention_map_" +
str(self.medcam_dict['counter']),
attention_map=attention_map,
heatmap=bool(raw_input is not None),
raw_input=raw_input)
self.medcam_dict['counter'] += 1
def _save_attention_map(self, attention_map, layer_output_dir, j, k,
raw_input):
"""Internal method for saving saving an attention map."""
if self.medcam_dict['save_pickle']:
self.medcam_dict['pickle_maps'].append(attention_map)
if self.medcam_dict['save_maps']:
medcam_utils.save_attention_map(
filename=layer_output_dir + "/attention_map_" +
str(self.medcam_dict['counter']) + "_" + str(j) + "_" + str(k),
attention_map=attention_map,
heatmap=self.medcam_dict['heatmap'],
raw_input=raw_input)
def save(attention_map, filename, heatmap, raw_input=None):
medcam_utils.save_attention_map(filename,
attention_map,
heatmap=heatmap,
raw_input=raw_input)
def save(attention_map, filename, raw_input=None):
medcam_utils.save_attention_map(filename, attention_map, heatmap=bool(raw_input is not None), raw_input=raw_input)
def save(attention_map, filename, heatmap):
medcam_utils.save_attention_map(filename, attention_map, heatmap)
