(rgb)
| 19 | return psnr |
| 20 | |
| 21 | def color_map_forward(rgb): |
| 22 | dim = rgb.shape[-1] |
| 23 | if dim==3: |
| 24 | return rgb.astype(np.float32)/255 |
| 25 | else: |
| 26 | rgb = rgb.astype(np.float32)/255 |
| 27 | rgb, alpha = rgb[:,:,:3], rgb[:,:,3:] |
| 28 | rgb = rgb * alpha + (1-alpha) |
| 29 | return rgb |
| 30 | |
| 31 | def main(): |
| 32 | parser = ArgumentParser() |