def print_meta_data(self):
"""
Print the meta data in a human readable format
:return:
"""
print(ersa_utils.make_center_string('=', 88, self.clc_name))
skip_keys = [
'gt_files', 'rgb_files', 'rgb_ext', 'gt_ext', 'file_ext', 'files'
]
for key, val in self.meta_data.items():
if key in skip_keys:
continue
if type(val) is list:
if len(val) >= 10:
print('{}: [{}, {}, ..., {}]'.format(
key, val[0], val[1], val[-1]))
else:
print('{}: {}'.format(key, val))
else:
print('{}: {}'.format(key, val))
print('Source file: {}'.format(' '.join([
'*{}*.{}'.format(ext1, ext2)
for ext1, ext2 in zip(self.rgb_ext, self.file_ext)
])))
if len(self.gt_ext) > 0:
print('GT file: {}'.format('*{}*.{}'.format(
self.gt_ext, self.file_ext[-1])))
print(ersa_utils.make_center_string('=', 88))
def load_results(self):
"""
load all the results computed by this process
:return: tile based iou, field based iou and overall iou
"""
print('Summary of results:')
result_name = os.path.join(self.score_save_dir, 'result.txt')
result = ersa_utils.load_file(result_name)
tile_dict, field_dict, overall = nn_utils.read_iou_from_file(result)
for key, val in field_dict.items():
field_str = ersa_utils.make_center_string(
'=', 50, '{}={:.2f}'.format(key, val * 100))
print(field_str)
for key_tile, val_tile in tile_dict.items():
if key in key_tile:
print('{}={:.2f}'.format(key_tile, val_tile * 100))
print(
ersa_utils.make_center_string(
'=', 50, 'Overall={:.2f}'.format(overall * 100)))
return tile_dict, field_dict, overall