pytorch 庫 pytorch 本身具有載入cifar10等數據集的函數,但是載入的是3*200*200的張量,當碰到要使用灰度圖像時,可以使用他本身的函數進行修改,以較快速的完成彩色圖像轉灰度圖
1.運行效果如圖所示2.實現代碼如下
億速云公眾號
手機網站二維碼
Copyright ? Yisu Cloud Ltd. All Rights Reserved. 2018 版權所有
廣州億速云計算有限公司粵ICP備17096448號 粵公網安備 44010402001142號增值電信業務經營許可證編號:B1-20181529