PyTorch/NumPy：从 rgb 图像创建二进制掩码答案

【问题标题】：PyTorch/NumPy: Create binary mask from rgb imagePyTorch/NumPy：从 rgb 图像创建二进制掩码
【发布时间】：2022-01-20 20:52:43
【问题描述】：

我有一个包含一批 4 个 RGB 128x128 图像的张量。所以张量的形状是 (4,128,128,3)。我需要从这个张量创建一个二进制掩码，如果图像是黑白的，每个像素都是黑色的，如果图像不是黑色的。我尝试了以下masks = torch.where(image > 0, 1.0, 0.)。但是这样得到的掩码显然仍然有三个通道。那么从 RGB 张量创建二进制掩码的最佳方法是什么？ numpy 也有同样的问题。

【问题讨论】：

标签： numpy pytorch

【解决方案1】：

我通过在调用where之前沿RGB通道应用sum解决了这个问题。

images_sum = images.sum(axis=3)
masks = torch.where(images_sum > 0, 1.0, 0.)

【讨论】：