| 15 | import pyvips |
| 16 | |
| 17 | def exportDMAPContent(dmap_path): |
| 18 | dmap = loadDMAP(dmap_path) |
| 19 | |
| 20 | basename = os.path.splitext(os.path.basename(dmap['file_name']))[0] |
| 21 | |
| 22 | pyvips.Image.new_from_array(np.uint8(dmap['depth_map'] * (1 / dmap['depth_max']) * 255)).write_to_file('%s_depth_map.png' % basename) |
| 23 | if dmap['has_normal']: |
| 24 | pyvips.Image.new_from_array(np.uint8((dmap['normal_map'] @ -dmap['R'] + 1) * 0.5 * 255)).write_to_file('%s_normal_map.png' % basename) |
| 25 | if dmap['has_conf']: |
| 26 | pyvips.Image.new_from_array(np.uint8(dmap['confidence_map'] * (1 / dmap['confidence_map'].max()) * 255)).write_to_file('%s_confidence_map.png' % basename) |
| 27 | |
| 28 | def main(): |
| 29 | parser = ArgumentParser() |