637ecf5877bbd6647caef2bd64925e66a586fb29,keras_rcnn/layers/losses/__init__.py,RCNN,regression_loss,#RCNN#,23

Before Change

        target_scores = self.target_scores[:, :, 1:]

        weights = keras.backend.sum(target_scores, axis=1)

        weights = keras.backend.sum(target_scores) / keras.backend.maximum(weights, keras.backend.epsilon())

        weights = 1.0 / (1 + keras.backend.exp(-weights))

        // mask out output values where class is different from targetrcnn loss
        // function
        a = keras_rcnn.backend.where(keras.backend.equal(target_scores, 1))

        a = keras.backend.cast(a, "int32")

        rr = a[:, :2]
        cc = a[:, 2:]

        indices = [
            keras.backend.concatenate([rr, cc * 4 + 0], 1),
            keras.backend.concatenate([rr, cc * 4 + 1], 1),
            keras.backend.concatenate([rr, cc * 4 + 2], 1),
            keras.backend.concatenate([rr, cc * 4 + 3], 1)
        ]

        indices = keras.backend.concatenate(indices, 0)

        weights = keras.backend.sum(target_scores * weights, axis=-1)
        weights = keras_rcnn.backend.gather_nd(weights, rr)

        weights = keras.backend.reshape(weights, (-1,))

        updates = keras.backend.tile(weights, (4,))

After Change

        updates = keras.backend.ones_like(indices, dtype=keras.backend.floatx())

        labels = keras_rcnn.backend.scatter_add_tensor(keras.backend.zeros_like(output_deltas, dtype="float32"), indices, updates[:, 0])

        loss = keras_rcnn.backend.smooth_l1(output_deltas * labels, target_deltas * labels, anchored=True)

Instances