import torch
rgb_boolean = rgb[:,:,0] == 1 #生成蒙版,数组为1的值为true,不为1的变为false
rgb[:,:,0] = rgb[:,:,0].masked_fill(rgb_boolean,255) #rgb中为1的变为255
python转换数组特定值的值
最新推荐文章于 2024-05-28 16:26:21 发布
import torch
rgb_boolean = rgb[:,:,0] == 1 #生成蒙版,数组为1的值为true,不为1的变为false
rgb[:,:,0] = rgb[:,:,0].masked_fill(rgb_boolean,255) #rgb中为1的变为255