使用numpy和opencv在python中裁剪基于2D掩模的3D图像

import numpy as np
import cv2

image_3d = np.load('path_to_3d_image.npy')

mask_2d = np.zeros((image_3d.shape[0], image_3d.shape[1]), dtype=np.uint8)
# 在掩模上绘制感兴趣区域，例如一个矩形
cv2.rectangle(mask_2d, (x1, y1), (x2, y2), (255), thickness=cv2.FILLED)

masked_image_3d = np.zeros_like(image_3d)
masked_image_3d[:, :, np.where(mask_2d)] = image_3d[:, :, np.where(mask_2d)]

np.save('path_to_save_masked_image.npy', masked_image_3d)