2017-09-11 11 views
0

私はcaffeで新しいです。私はhttps://github.com/shelhamer/fcn.berkeleyvision.orgからnyud-fcn32s-color-dを訓練しようとしています。私はソルバーの定義とトレーニング段階の前にトレーニングデータを管理する方法を理解していません。私はhttp://cs.nyu.edu/~silberman/datasets/nyu_depth_v2.htmlからnyud-v2 .matデータセットをダウンロードしましたが、nyud_layer(python layer)をどのようにフィードするのか分かりません。申し訳ありませんが、私は初心者ですが、私はこのトレーニングをすることが重要ですので、どんな助けも非常に高く評価されるでしょう。フィードトレーニングデータのpython層(FCN CAFFE)

答えて

1

あなたが電車/ valが

layer { 
    name: "color" 
    type: "Python" 
    top: "color" 
    top: "hha" 
    top: "label" 
    python_param { 
    module: "nyud_layers" 
    layer: "NYUDSegDataLayer" 
    param_str: "{\'tops\': [\'color\', \'hha\', \'label\'], \'seed\': 1337, \'nyud_dir\': \'../data/nyud\', \'split\': \'test\'}" 
    } 
} 

として定義pythonの層を、prototxt TOUを追加し、in the implementationが説明するように、正しいのparamsを埋める必要があります。

- nyud_dir: path to NYUDv2 dir 
- split: train/val/test 
- tops: list of tops to output from {color, depth, hha, label} 
- randomize: load in random order (default: True) 
- seed: seed for randomization (default: None/current time) 
+0

answer..Iためのおかげで」 trainval.protoxtがうまく動かなかったので、nyud_layerで何かを変更する必要がありますか?そして、nyud_dirには何が必要ですか?私はクラス、列車、val、trainvalとtest(.txt)を持っていますが、nyu_depth_v2_labeled.matを追加しましたが、このファイルだけではソルバーを作成できません。 – Gio

+0

nyud_dirをNYUDデータセット(たとえば../data/nyud)を解凍した場所に設定し、希望するトップのリストを選択する必要があります。あなたの質問を編集してsolver.txtの内容とエラーを含めることはできますか? – rkellerm

+0

本当にありがとうございます。私は別の質問があります。現時点で私はソルバーを定義することができますが、想像していたように、私の2GBのGPUを使って、solver.net.forward()トレーニングを開始しようとすると、メモリ不足でキューが消えてしまいます。どのように状況を管理できますか?バッチサイズを減らすためのnyud_layerの方法がありますか? data/nyud/dataには、深さ、hha、画像、生の深度フォルダ(.pngファイル(560x425))があります。 – Gio

関連する問題