当前位置:主页 > 软件编程 > Python代码 >

Pytorch 之修改Tensor部分值方式

时间:2021-02-17 14:03:26 | 栏目:Python代码 | 点击:

一:背景引入

对于一张图片,怎样修改局部像素值?

二:利用Tensor方法

比如输入全零tensor,可认为为黑色图片

>>> n=torch.FloatTensor(3,3,4).fill_(0)
>>> n
    tensor([[[0., 0., 0., 0.], 
         [0., 0., 0., 0.],                          
         [0., 0., 0., 0.]], 
         
         [[0., 0., 0., 0.], 
         [0., 0., 0., 0.],
         [0., 0., 0., 0.]], 
 
         [[0., 0., 0., 0.], 
         [0., 0., 0., 0.],
         [0., 0., 0., 0.]]])     

现在修改部分值:

>>> n[:,0:2,1:3]=1

结果为

>>> n
    tensor([[[0., 1., 1., 0.], 
         [0., 1., 1., 0.],                          
         [0., 0., 0., 0.]], 
         
         [[0., 1., 1., 0.], 
         [0., 1., 1., 0.],
         [0., 0., 0., 0.]], 
 
         [[0., 1., 1., 0.], 
         [0., 1., 1., 0.],
         [0., 0., 0., 0.]]]) 

可以看到利用切片的思想可以修改tensor中的部分值。这样修改后,黑色图片上就会有白色的小洞洞,因为在tensor中,1代表是白色,RGB中的255。

您可能感兴趣的文章:

相关文章