Exemplo n.º 1
0
 def add_div(self, input_names, output_name):
     layer = caffe_net.LayerParameter(name=output_name, type='Eltwise',
                 bottom=input_names,
                 top=[output_name])
     layer.eltwise_param(3) #Div
     self.caffe_model.add_layer(layer)
     data = []
     for name in input_names:
         data.append(self.data_dict[name])
     self.data_dict[output_name] = Operators.divide(self.data_dict[input_names[0]], self.data_dict[input_names[1]], output_name)
     return output_name