Wie concate Schicht erzeugen prototxt Python mit

Question

ich prototxt haben wie folgt:

layer { 
    name: "data" 
    type: "HDF5Data" 
    top: "data1" 
    top: "data2" 
    top: "label" 
    include { 
    phase: TRAIN 
    } 
    hdf5_data_param { 
    source: "./source_list.txt" 
    batch_size: 2 
    shuffle: true 
    } 
} 
layer { 
    name: "concat" 
    type: "Concat" 
    bottom: "data1" 
    bottom: "data2" 
    top: "data" 
    concat_param { 
    concat_dim:1 
    } 
} 

ich oben prototxt mit caffe NetSpec in Python generieren möchten. Es war jedoch falsch. Das ist mein Code. Bitte hilf mir, es zu beheben. Dank

from caffe import layers as L 
... 
n = caffe.NetSpec() 
n.data, n.label = L.HDF5Data(top=["data1", "data2"], batch_size=2, 
          source="./source_list.txt", ntop=2,shuffle= True, 
          include={'phase': caffe.TRAIN}) 
n.concat = L.Concat(n.data["data1"], n.data["data2"]) 

Answer 1

Sie benötigen zwei data Ausgänge jetzt

n.data1, n.data2, n.label = L.HDF5Data(ntop=3, name="data", 
             hdf5_data_param={'source': "./source_list.txt", 
                 'shuffle': True, 
                 'batch_size': 2} 
             include={'phase': caffe.TRAIN}) 
n.data = L.Concat(n.data1, n.data2, name="concat", concat_param={'concat_dim':1})