gpt4 book ai didi

python - 如何在同一 prototxt 中生成用于训练和测试的数据层(HDF5)?

转载 作者:太空宇宙 更新时间:2023-11-03 15:39:28 25 4
gpt4 key购买 nike

我有一个 HDF5 类型的数据层。它包含预期的训练和测试阶段

name: "LogisticRegressionNet"
layer {
name: "data"
type: "HDF5Data"
top: "data"
top: "label"
include {
phase: TRAIN
}
hdf5_data_param {
source: "examples/hdf5_classification/data/train.txt"
batch_size: 10
}
}
layer {
name: "data"
type: "HDF5Data"
top: "data"
top: "label"
include {
phase: TEST
}
hdf5_data_param {
source: "examples/hdf5_classification/data/test.txt"
batch_size: 10
}
}

我想用python来生成它。这是我的代码

import caffe
from caffe import layers as L # pseudo module using __getattr__ magic to generate protobuf messages
from caffe import params as P # pseudo module using __getattr__ magic to generate protobuf messages
n = caffe.NetSpec()
n.data, n.label = L.HDF5Data(batch_size=batch_size, source='examples/hdf5_classification/data/train.txt', ntop=2, include={'phase': caffe.TRAIN})
n.data, n.label = L.HDF5Data(batch_size=batch_size, source='examples/hdf5_classification/data/test.txt',ntop=2, include={'phase': caffe.TEST})

但是,我的输出只是测试阶段。我该如何修复它?谢谢

layer {
name: "data"
type: "HDF5Data"
top: "data"
top: "label"
include {
phase: TEST
}
hdf5_data_param {
source: "examples/hdf5_classification/data/test.txt"
batch_size: 2
}
}

最佳答案

这是一个悬而未决的问题in caffe (您可以在那里找到其他相关 SO 线程的链接)。

你可以做的是编写两个prototxt,一个用于训练,一个用于测试。 solver.prototxt支持定义训练网文件名和测试网文件名。

关于python - 如何在同一 prototxt 中生成用于训练和测试的数据层(HDF5)?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42289880/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com