2

caffe net でデータ層を定義するために作成した LMDB ファイルを使用しようとしていますが、以下のエラーが発生します

TypeError: 'LMDB' has type (type 'str'), but expected one of: (type 'int', type 'long')

lmdb ファイル ( caffe/build/tools/convert_imageset) を生成するスクリプトに渡したテキスト ファイルのラベルを確認しました。
ここで何か不足していますか？

編集-1：これが私のデータレイヤー定義です：

n.data,n.labels = L.Data(batch_size = batch_size, 
                         source=lmdb_src, 
                         backend = "LMDB", 
                         transform_param = dict(mean_file = mean_file),
                         ntop=2)
4

1 に答える 1

3

設定しようとしています

backend: "LMDB"

代わりに、ネット定義で

backend: LMDB

LMDBは文字列として渡されるのではなく、列挙された整数として渡されることに注意してください。

やるべきことは決まっている

backend = caffe.Data.LMDB

caffe protobuff 定義で設定された列挙値を使用します。

于 2016-05-31T13:55:30.913 に答える