如何将多个文件加载到caffe模型

时间:2017-02-12 15:26:12

标签: neural-network deep-learning caffe hdf5

我正在尝试使用以下caffe模型进行预测" https://github.com/amandajshao/www_deep_crowd"。这个模型不作为输入图像,但它需要一些运动和外观功能。我将每个帧的这些功能作为.mat文件并将它们转换为hdf5。 但我无法弄清楚如何将它们加载到咖啡馆。 有什么帮助吗?

1 个答案:

答案 0 :(得分:0)

转换为HDF5的方式超过了其中的一半......

"HDF5Data"图层可以为hdf5文件中存储的每个数据集生成多个"top" s:1。所以,假设你used matlab to write your hdf5 files

hdf5write('my_data.h5', 'motion', single( motion )  );
hdf5write('my_data.h5', 'appearance', single(appearnace), 'WriteMode', 'append' );

然后,您可以拥有一个"HDF5Data"图层,其中包含两个"top“s:

layer {
  name: "data"
  type: "HDF5Data"
  top: "motion"
  top: "appearance"
  hdf5_data_param { source: "path/to/hdf_list.txt" }
}

您可能希望密切关注数据的存储顺序,因为Matlab以“Fortran”方式存储数组,而caffe期望其数据以“C”方式排序。请参阅this post for more information