深度學(xué)習(xí)中,數(shù)據(jù)是基礎(chǔ)。對(duì)于自己的數(shù)據(jù),為了方便的應(yīng)用開源庫(kù)Caffe和tensorflow,可以將數(shù)據(jù)轉(zhuǎn)成其需要的格式,比如LMDB和tfrecords。當(dāng)然還有很多其他的方式,比如對(duì)于Caffe可以直接讀取圖像格式數(shù)據(jù),hdf5格式等,甚至可以自己寫一個(gè)python的DataLayer,但有時(shí)候由于各種原因,仍然需要我們將數(shù)據(jù)轉(zhuǎn)化為L(zhǎng)MDB格式。
今天分享一段很簡(jiǎn)單的數(shù)據(jù)轉(zhuǎn)換代碼,如下:
首先我們先寫這個(gè)數(shù)據(jù)轉(zhuǎn)換類:


然后我們就可以調(diào)用這個(gè)類來(lái)處理數(shù)據(jù)了,以下生成一些假數(shù)據(jù),做個(gè)測(cè)試:


抱歉簡(jiǎn)書上傳代碼不太方便,粘貼過(guò)來(lái)格式很難看,大家就自己寫一遍,更加有成就感。