劃分訓練集、測試集,製作自己的數據集

從文件路徑讀取圖片,將圖片的數組存爲npz格式。 數據集:人臉卡通表情FERG數據,包含6個卡通人物,每個卡通人物7個表情,每張圖片256*256的png格式,數據集壓縮包大小2.95GB。 圖片路徑tree: 數據集的85%爲訓練集,15%爲測試集 代碼如下: ''' id存索引,ep存索引,image存矩陣 將圖片預處理爲numpy數組,存儲於.npz ''' import time fro
相關文章
相關標籤/搜索