pytorch 庫 pytorch 本身具有載入cifar10等數據集的函數,但是載入的是3*200*200的張量,當碰到要使用灰度圖像時,可以使用他本身的函數進行修改,以較快速的完成彩色圖像轉灰度圖
1.運行效果如圖所示2.實現代碼如下