為了使用自己的圖像數據，需要仿照pytorch數據輸入創建新的類，其中數據格式為numpy.ndarray。 將自己的圖片保存到numpy.ndarray中，然后創建類 from torch.ut
億速云公眾號
手機網站二維碼
Copyright ? Yisu Cloud Ltd. All Rights Reserved. 2018 版權所有
廣州億速云計算有限公司粵ICP備17096448號 粵公網安備 44010402001142號增值電信業務經營許可證編號：B1-20181529
