check_images.py

import matplotlib.pyplot as plt
import numpy as np


def main():
    img = np.load("img.npy")
    img = np.transpose(img, (1, 2, 0))
    img = img.astype(np.uint8)
    new_img = np.load("new_img.npy")
    print(new_img.shape, new_img.min(), new_img.max())
    new_img = np.transpose(new_img, (1, 2, 0))
    new_img = new_img.astype(np.uint8)
    fig, axs = plt.subplots(nrows=1, ncols=2)
    axs[0].imshow(img)
    axs[1].imshow(new_img)
    plt.get_current_fig_manager().window.showMaximized()
    plt.show()


if __name__ == "__main__":
    main()