【Python】RGB画像・デプス画像を正規化して色の変化を大きくする(torch・numpy)

import torch
import numpy as np
from PIL import Image
from skimage import io
from torchvision.utils import save_image
import matplotlib.pyplot as plt

# 奥行き画像を読み込み
depth_img = io.imread("image.png")

# PytorchのTensorに変換
depth_tensor = torch.tensor(depth_img)

# 最大値から引く値
a = 10

# デプス値の最小値と最大値を求める
min_depth = depth_tensor.min().item()
max_depth = depth_tensor.max().item()-a

# デプス値を0～1にスケーリング
scaled_depth = (depth_tensor - min_depth) / (max_depth - min_depth)
scaled_depth[scaled_depth > 1] = 1

# 結果を表示
# print("Min depth: ", min_depth)
# print("Max depth: ", max_depth)
# print("Scaled depth: ", scaled_depth)

plt.imshow(scaled_depth, cmap='gray')
plt.savefig('torch_image.png', bbox_inches='tight')

min_depth = depth_tensor.min().item()+100
max_depth = depth_tensor.max().item()-55

scaled_depth[scaled_depth > 1] = 1
scaled_depth[scaled_depth < 0] = 0