pytorch 庫
pytorch 本身具有載入cifar10等數據集的函數，但是載入的是3*200*200的張量，當碰到要使用灰度圖像時，可以使用他本身的函數進行修改，以較快速的完成彩色圖像轉灰度圖像
pytorch函數
dataset = dset.CIFAR10(root='../train/data', download=True,transform=transforms.Compose([
transforms.Scale(200),
transforms.ToTensor(),
transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
]))
這里包含了對圖像進行的一些處理，尺寸的控制，Normalize等
修改pytorch 庫函數
找到transforms 的庫函數位置，在后面添加 類 Gray（）
class Gray(object) def __call__(self, tensor): # TODO: make efficient R = tensor[0] G = tensor[1] B = tensor[2] tensor[0]=0.299*R+0.587*G+0.114*B tensor = tensor[0] tensor = tensor.view(1,200,200) return tensor
tensor.view 這里使用是為了將tensor 做成 1*200*200的大小，否則出來是200*200
以上這篇pytorch 彩色圖像轉灰度圖像實例就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持億速云。
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
