third_party/DarkFeat/datasets/InvISP/utils/commons.py

import numpy as np


def denorm(img, max_value):
    img = img * float(max_value)
    return img


def preprocess_test_patch(input_image, target_image, gt_image):
    input_patch_list = []
    target_patch_list = []
    gt_patch_list = []
    H = input_image.shape[2]
    W = input_image.shape[3]
    for i in range(3):
        for j in range(3):
            input_patch = input_image[
                :,
                :,
                int(i * H / 3) : int((i + 1) * H / 3),
                int(j * W / 3) : int((j + 1) * W / 3),
            ]
            target_patch = target_image[
                :,
                :,
                int(i * H / 3) : int((i + 1) * H / 3),
                int(j * W / 3) : int((j + 1) * W / 3),
            ]
            gt_patch = gt_image[
                :,
                :,
                int(i * H / 3) : int((i + 1) * H / 3),
                int(j * W / 3) : int((j + 1) * W / 3),
            ]
            input_patch_list.append(input_patch)
            target_patch_list.append(target_patch)
            gt_patch_list.append(gt_patch)

    return input_patch_list, target_patch_list, gt_patch_list