PROSAGA码农传奇-测试训练-如何在caffe中训练/测试我自己的数据集？

<div class =“post-text”itemprop =“text”>
  
    我认为将数据从Matlab传输到caffe的最直接的方法是通过HDF5文件。
  
  
    首先，使用Matlab将数据保存在HDF5文件中
    <a href="http://www.mathworks.com/help/matlab/ref/hdf5write.html" rel="nofollow noreferrer">
       <code>
 hdf5write
 </code>
    </A>
    。我假设您的训练数据存储在变量名称中
     <code>
 X
 </code>
     尺寸为300×30，标签存放在
     <code>
 y
 </code>
     一个300乘1的向量：
  
   <pre>
 <code>
 hdf5write('my_data.h5', '/X', 
 single( permute(reshape(X,[300, 30, 1, 1]),[4:-1:1]) ) );
hdf5write('my_data.h5', '/label', 
 single( permute(reshape(y,[300, 1, 1, 1]),[4:-1:1]) ), 
 'WriteMode', 'append' );

</code>
 </pre>
  
    请注意，数据保存为4D数组：第一个维度是要素数量，第二个维度是要素的维度，后两个是1（表示无空间维度）。另请注意，HDF5中的数据名称是
     <code>
 "X"
 </code>
     和
     <code>
 "label"
 </code>
      - 这些名称应该用作
     <code>
 "top"
 </code>
     输入数据层的blob。
  
  
    为什么
     <code>
 permute
 </code>
    ？请参阅
    <a href="https://stackoverflow.com/a/34431532/1714410">
      这个答案
    </A>
     作出解释。
  
  
    您还需要准备一个文本文件，列出您正在使用的所有hdf5文件的名称（仅在您的情况下）
     <code>
 my_data.h5
 </code>
    ）。文件
     <code>
 /path/to/list/file.txt
 </code>
     应该有一条线
  
  <BLOCKQUOTE>
    
      /path/to/my_data.h5
    
  </BLOCKQUOTE>
  
    现在，您可以将输入数据层添加到train_val.prototxt
  
   <pre>
 <code>
 layer {
 type: "HDF5Data"
 name: "data"
 top: "X" # note: same name as in HDF5
 top: "label" # 
 hdf5_data_param {
 source: "/path/to/list/file.txt"
 batch_size: 20
 }
 include { phase: TRAIN }
}

</code>
 </pre>
  
    有关hdf5输入图层的更多信息，请参阅
    <a href="https://stackoverflow.com/a/31634295/1714410">
      这个答案
    </A>
    。
  
</DIV>