depth_npy_to_map.py

"""
Author: Yanxiu Jin
Date: 2025-03-17
Description: Convert depth_map.npy to grayscale png,
it can also be used for any type of npy files like saliency.npy

"""
import numpy as np
import matplotlib.pyplot as plt
import os
import imageio
import glob
from skimage import exposure


local_dir = "D:\\2021-han-scene-simplification-master\\2021-han-scene-simplification-master"
output_frames_dir = local_dir+"\\depth_output_npy\\kitchen20fps_TCMono"
all_frames = glob.glob(output_frames_dir+"\\*.jpg")

for count in np.arange(1, len(all_frames)+1 ):  # each frame !!modified +1
    f_name = output_frames_dir+"\\frame_%03d.npy" % count

    depth_map = np.load(f_name)  # Shape: (H, W)

    # depth_map_eq = exposure.equalize_hist(depth_map)

    plt.imshow(depth_map, cmap='gray')
    plt.axis('off')  # Hide axis

    depth_norm = (depth_map - depth_map.min()) / (depth_map.max() - depth_map.min()) * 255
    depth_uint8 = depth_norm.astype(np.uint8)
    mask_out_dir = local_dir + "\\depth_output_npy\\kitchen20fps_TCMono_frames"

    if not os.path.exists(mask_out_dir):
        os.makedirs(mask_out_dir)
    depth_filename = os.path.join(mask_out_dir, f"frame_{count:03d}_depth.png")
    imageio.imwrite(depth_filename, depth_uint8)

    plt.savefig(depth_filename, bbox_inches='tight', pad_inches=0)
    plt.show()