首頁(yè) 猿問(wèn) 回歸caffe的測(cè)試標(biāo)簽，不允許浮動(dòng)？

回歸caffe的測(cè)試標(biāo)簽，不允許浮動(dòng)？

源碼算法與數(shù)據(jù)結(jié)構(gòu)

慕仙森 2019-08-19 16:55:30

回歸caffe的測(cè)試標(biāo)簽，不允許浮動(dòng)？我正在使用caffe進(jìn)行回歸，而我test.txt和train.txt文件是這樣的：/home/foo/caffe/data/finetune/flickr/3860781056.jpg 2.0 /home/foo/caffe/data/finetune/flickr/4559004485.jpg 3.6 /home/foo/caffe/data/finetune/flickr/3208038920.jpg 3.2 /home/foo/caffe/data/finetune/flickr/6170430622.jpg 4.0 /home/foo/caffe/data/finetune/flickr/7508671542.jpg 2.7272我的問(wèn)題是，當(dāng)我在閱讀時(shí)使用浮動(dòng)標(biāo)簽時(shí)，似乎caffe不允許像2.0這樣的浮動(dòng)標(biāo)簽，例如'test.txt'文件caffe只能識(shí)別共1張圖片這是錯(cuò)的。但是當(dāng)我例如將文件中的2.0更改為2并且以下行相同時(shí)，caffe現(xiàn)在給出了共2張圖片暗示浮動(dòng)標(biāo)簽是造成問(wèn)題的原因。任何人都可以幫助我，解決這個(gè)問(wèn)題，我肯定需要使用浮動(dòng)標(biāo)簽進(jìn)行回歸，所以有人知道解決方案或解決方案嗎？提前致謝。編輯對(duì)于任何面臨類(lèi)似問(wèn)題的人來(lái)說(shuō)，使用caffe來(lái)訓(xùn)練Lenet的CSV數(shù)據(jù)可能會(huì)有所幫助。感謝@Shai。

查看完整描述

3 回答

侃侃爾雅

TA貢獻(xiàn)1801條經(jīng)驗(yàn) 獲得超16個(gè)贊

如果LMDB是必需/首選的，這里有一個(gè)關(guān)于如何從float數(shù)據(jù)創(chuàng)建LMDB的片段（改編自這個(gè)?github注釋?zhuān)?/p>

import?lmdbimport?caffedef?scalars_to_lmdb(scalars,?path_dst):
????db?=?lmdb.open(path_dst,?map_size=int(1e12))
????with?db.begin(write=True)?as?in_txn:????
????????for?idx,?x?in?enumerate(scalars):????????????
????????????content_field?=?np.array([x])
????????????#?get?shape?(1,1,1)
????????????content_field?=?np.expand_dims(content_field,?axis=0)
????????????content_field?=?np.expand_dims(content_field,?axis=0)
????????????content_field?=?content_field.astype(float)
????????????dat?=?caffe.io.array_to_datum(content_field)
????????????in_txn.put('{:0>10d}'.format(idx)?dat.SerializeToString())
????db.close()

反對(duì) 回復(fù) 2019-08-19