Pytorch 之修改Tensor部分值方式
一:背景引入
對于一張圖片,怎樣修改局部像素值?
二:利用Tensor方法
比如輸入全零tensor,可認(rèn)為為黑色圖片
>>> 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.]]])
現(xiàn)在修改部分值:
>>> n[:,0:2,1:3]=1
結(jié)果為
>>> 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中的部分值。這樣修改后,黑色圖片上就會(huì)有白色的小洞洞,因?yàn)樵趖ensor中,1代表是白色,RGB中的255。
以上這篇Pytorch 之修改Tensor部分值方式就是小編分享給大家的全部內(nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
在Mac OS上使用mod_wsgi連接Python與Apache服務(wù)器
這篇文章主要介紹了在Mac OS上使用mod_wsgi連接Python與Apache服務(wù)器的方法,同時(shí)文中還介紹了使用Python的Django框架時(shí)mod_wsgi連接方式下可能遇到的問題的一般解決方法,需要的朋友可以參考下2015-12-12淺談django model的get和filter方法的區(qū)別(必看篇)
下面小編就為大家?guī)硪黄獪\談django model的get和filter方法的區(qū)別(必看篇)。小編覺得挺不錯(cuò)的,現(xiàn)在就分享給大家,也給大家做個(gè)參考。一起跟隨小編過來看看吧2017-05-05Python函數(shù)中apply、map、applymap的區(qū)別
這篇文章主要介紹了 Python函數(shù)中apply、map、applymap的區(qū)別 ,文章圍繞 Python函數(shù)中apply、map、applymap的相關(guān)資料展開詳細(xì)內(nèi)容,需要的朋友可以參考一下2021-11-11詳解Python中的__getitem__方法與slice對象的切片操作
Python中想要使類的實(shí)例像list一樣使用下標(biāo),可以用__getitem__方法,而配合slice對象則可以實(shí)現(xiàn)list一樣的切片,詳解Python中的__getitem__方法與slice對象的切片操作2016-06-06