Exemplo n.º 1
0
 def add_input(self, output_name, input_shape):
     layer = caffe_net.LayerParameter(name=output_name, type='Input',
                                   top=[output_name])
     layer.input_param(input_shape)
     self.caffe_model.add_layer(layer)
     self.data_dict[output_name] = Operators.get_input(output_name, input_shape)