numpy数组替换其中的值（如1替换为255）

*Major*

已于 2023-07-06 13:44:20 修改

阅读量675

点赞数 1

文章标签： numpy

于 2023-07-06 13:38:41 首次发布

本文链接：https://blog.csdn.net/qq_41375318/article/details/131574752

版权

numpy数组替换其中的值

data[data==1] = 9999

实例

import numpy as np
data = np.array([[1,2,3],[1,5,6]])
data[data==1] = 9999

在这里插入图片描述

data == 9999

在这里插入图片描述

1替换为255

mask1[mask1 == 1] = 255

在这里插入图片描述

 masks = Image.open(os.path.join(self.data_dir, 'SegmentationObject/{}.png'.format("2007_000876")))  # mask的数值来自于图片
 # 压缩到0-1之间
 masks = transforms.ToTensor()(masks)
 # 找到mask中有哪几种不同的数值
 uni = masks.unique()
 # 其中在0和1之间的数值分别代表一个mask（实例分割）
 uni = uni[(uni > 0) & (uni < 1)]
 # 将图片中等于uni的部分分别转为值为0,1的图片
 res = uni.reshape(-1,1,1)
 masks = (masks == uni.reshape(-1, 1, 1)).to(torch.uint8)

 mask1 = masks.cuda().cpu().numpy()[0]
 mask2 = masks.cuda().cpu().numpy()[1]
 mask1[mask1 == 1] = 255
 m1 = Image.fromarray(mask1)
 m1.show()

 mask2[mask1 == 1] = 255
 m2 = Image.fromarray(mask2)
 m2.show()