| ######################################################################### | |
| # Copyright (c) 2020, NVIDIA CORPORATION. All rights reserved. | |
| # | |
| # Redistribution and use in source and binary forms, with or without | |
| # modification, are permitted provided that the following conditions | |
| # are met: | |
| # * Redistributions of source code must retain the above copyright | |
| # notice, this list of conditions and the following disclaimer. | |
| # * Redistributions in binary form must reproduce the above copyright | |
| # notice, this list of conditions and the following disclaimer in the | |
| # documentation and/or other materials provided with the distribution. | |
| # * Neither the name of NVIDIA CORPORATION nor the names of its | |
| # contributors may be used to endorse or promote products derived | |
| # from this software without specific prior written permission. | |
| # | |
| # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ``AS IS'' AND ANY | |
| # EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | |
| # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR | |
| # PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR | |
| # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, | |
| # EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, | |
| # PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR | |
| # PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY | |
| # OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | |
| # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | |
| # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
| ######################################################################### | |
| # FLIP: A Difference Evaluator for Alternating Images | |
| # High Performance Graphics, 2020. | |
| # by Pontus Andersson, Jim Nilsson, Tomas Akenine-Moller, Magnus Oskarsson, Kalle Astrom, and Mark D. Fairchild | |
| # | |
| # Pointer to our paper: https://research.nvidia.com/publication/2020-07_FLIP | |
| # code by Pontus Andersson, Jim Nilsson, and Tomas Akenine-Moller | |
| import numpy as np | |
| from flip import compute_flip | |
| from utils import * | |
| if __name__ == '__main__': | |
| # Set viewing conditions | |
| monitor_distance = 0.7 | |
| monitor_width = 0.7 | |
| monitor_resolution_x = 3840 | |
| # Compute number of pixels per degree of visual angle | |
| pixels_per_degree = monitor_distance * (monitor_resolution_x / monitor_width) * (np.pi / 180) | |
| # Load sRGB images | |
| reference = load_image_array('../images/reference.png') | |
| test = load_image_array('../images/test.png') | |
| # Compute FLIP map | |
| deltaE = compute_flip(reference, test, pixels_per_degree) | |
| # Save error map | |
| index_map = np.floor(255.0 * deltaE.squeeze(0)) | |
| use_color_map = True | |
| if use_color_map: | |
| result = CHWtoHWC(index2color(index_map, get_magma_map())) | |
| else: | |
| result = index_map / 255.0 | |
| save_image("../images/flip.png", result) | |
| print("Done") | |