Caffe: como obter a fase de uma camada Python?
Eu criei um"Python"
camada"myLayer"
no café e use-o na redetrain_val.prototxt
Eu insiro a camada assim:
layer {
name: "my_py_layer"
type: "Python"
bottom: "in"
top: "out"
python_param {
module: "my_module_name"
layer: "myLayer"
}
include { phase: TRAIN } # THIS IS THE TRICKY PART!
}
Agora, minha camada só participa doTRAIN
fase inicial da rede,
como posso saber que na minha camadasetup
função??
class myLayer(caffe.Layer):
def setup(self, bottom, top):
# I want to know here what is the phase?!!
...
PS,
Publiquei esta pergunta emGrupo do Google "Usuários do Caffe" também. Vou atualizar se algo aparecer lá.