eb737460e317877af263183245ebac792c02aa20,keras_retinanet/layers/_misc.py,NonMaximumSuppression,call,#NonMaximumSuppression#Any#,121

Before Change

    def call(self, inputs, **kwargs):
        boxes, classification, detections = inputs

        boxes          = keras.backend.reshape(boxes, (-1, 4))
        classification = keras.backend.reshape(classification, (-1, self.num_classes))

        scores  = keras.backend.max(classification, axis=1)
        labels  = keras.backend.argmax(classification, axis=1)

After Change

        boxes, classification, detections = inputs

        // TODO: support batch size > 1.
        boxes          = boxes[0]
        classification = classification[0]
        detections     = detections[0]

        scores  = keras.backend.max(classification, axis=1)

Instances