1. 程式人生 > >triplet資料集載入

triplet資料集載入

步驟如下:

           A:首先建立標籤資料匹配,格式為txt,每一行有三個標籤(Anchor,Positive,Negative),總共五千行,具體步驟可以參考我的另一篇部落格https://blog.csdn.net/qq_33824968/article/details/84861777,txt文字在下載欄檢視。

          B:開始在工程中建立.py檔案,定義一個class TripletImageLoaderCifar10(Dataset),以原來的Cifar10資料呼叫為母本,然後分塊編寫即可,分一下幾個小步驟
                        1 # 定義data_path 聯合path定義出兩個txt資料夾,self定義transform 這樣給定的四個引數都設定完畢

                        2 # 開啟檔案,把txt中的影象名字調出來

                       3 # 開啟triplet標籤資料集,將每一行資訊放入list中

以上三個步驟完成基本上完成了

貼一下程式碼。