首頁猿問回歸caffe的測試標簽，不允許浮動？

回歸caffe的測試標簽，不允許浮動？

源碼算法與數(shù)據(jù)結構

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

回歸caffe的測試標簽，不允許浮動？我正在使用caffe進行回歸，而我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我的問題是，當我在閱讀時使用浮動標簽時，似乎caffe不允許像2.0這樣的浮動標簽，例如'test.txt'文件caffe只能識別共1張圖片這是錯的。但是當我例如將文件中的2.0更改為2并且以下行相同時，caffe現(xiàn)在給出了共2張圖片暗示浮動標簽是造成問題的原因。任何人都可以幫助我，解決這個問題，我肯定需要使用浮動標簽進行回歸，所以有人知道解決方案或解決方案嗎？提前致謝。編輯對于任何面臨類似問題的人來說，使用caffe來訓練Lenet的CSV數(shù)據(jù)可能會有所幫助。感謝@Shai。

查看完整描述

3 回答

侃侃爾雅

TA貢獻1801條經(jīng)驗獲得超16個贊

如果LMDB是必需/首選的，這里有一個關于如何從float數(shù)據(jù)創(chuàng)建LMDB的片段（改編自這個?github注釋）：

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()

反對回復 2019-08-19