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)