如何从caffe中获取hdf5输入的输出?

时间:2016-10-12 07:23:12

标签: matlab caffe hdf5

我使用hdf5文件

提供下面定义的网络
    name: "match_layers"
    layers {
      name: "data"
      type: HDF5_DATA
      top: "feat1"
      top: "feat2"
      hdf5_data_param {
        source: "vgg16_train_test_pair.h5list"
        batch_size: 50
      }
    }

    layers {
      name: "concat"
      bottom: "feat1"
      bottom: "feat2"
      top: "union"
      type: CONCAT
        concat_param{
        concat_dim: 1
      }
    }  

    layers {
      bottom: "union"
      top: "fc1"
      name: "fc1"
      type: INNER_PRODUCT
      inner_product_param {
      num_output: 1024
    }

    .......

vgg16_train_test_pair.h5list中的内容是:

     vgg16_train_test_pair.h5

并且hdf5文件已经退出,我想使用matlab接口获取 fc1 的输出,我试过了:

     net = caffe.Net(model, weights, 'test');
     a = net.forward();
     fc1 = net.blobs('fc1').get_data();

但它没有用,有人可以帮我解决吗?

0 个答案:

没有答案