Exemplo n.º 1
0
 def add_check(self, left_name, right_name, condition, status_name):
     layer = caffe_net.LayerParameter(name=status_name, type='Check',
                 bottom=[left_name, right_name], top=[status_name])
     layer.check_param(condition)
     self.caffe_model.add_layer(layer)
     self.data_dict[status_name] = Operators.check(self.data_dict[left_name],
                                       self.data_dict[right_name],
                                       condition,
                                       status_name)
     return status_name