网络编程
位置:首页>> 网络编程>> Python编程>> Pytorch 之修改Tensor部分值方式

Pytorch 之修改Tensor部分值方式

作者:my-GRIT  发布时间:2023-04-11 06:45:12 

标签:Pytorch,Tensor

一:背景引入

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

二:利用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。

来源:https://blog.csdn.net/qq_39709535/article/details/82720938

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com