9a062c6473109dfdd545d1db84f62b5037c53128,snntoolbox/model_libs/caffe_input_lib.py,ModelParser,get_outbound_layers,#ModelParser#Any#,124

Before Change

        layers = self.get_layer_iterable()
        outbound = []
        for outb in layer.top:  // Contains only labels
            for layer in layers:
                if outb == layer.name:
                    outbound.append(layer)
                    break
        return outbound

    def parse_concatenate(self, layer, attributes):
        attributes.update({"mode": "concat",

After Change

    def get_outbound_layers(self, layer):
        layers = self.get_layer_iterable()
        layer_ids = [id(l) for l in layers]
        current_idx = layer_ids.index(id(layer))
        return [] if current_idx + 1 >= len(layer_ids) \
            else [layers[current_idx + 1]]

    def parse_concatenate(self, layer, attributes):
        attributes.update({"mode": "concat",
                           "concat_axis": layer.concat_param.axis})

Instances