eb4732369125efb8c9c87ad79544e99c1aa3b67e,keras_rcnn/applications/_jhung2019.py,JHung2019,predict,#JHung2019#Any#Any#Any#Any#,294

Before Change

        self.load_weights("JHung2019.hdf5", by_name=True)

    def predict(self, x, batch_size=None, verbose=0, steps=None):
        target_bounding_boxes = numpy.zeros((x.shape[0], 1, 4))

        target_categories = numpy.zeros((x.shape[0], 1, self.n_categories))

        target_metadata = numpy.array([[x.shape[1], x.shape[2], 1.0]])

        x = [
            target_bounding_boxes,
            target_categories,
            x,
            target_metadata
        ]

        return super(JHung2019, self).predict(x, batch_size, verbose, steps)

After Change

        predicted_bounding_boxes, predicted_categories = prediction

        predicted_bounding_boxes = numpy.squeeze(predicted_bounding_boxes)

        predicted_categories = numpy.squeeze(predicted_categories)

        predicted_categories = numpy.argmax(predicted_categories, axis=-1)

        indices = numpy.where(predicted_categories > 0)

        predicted_bounding_boxes = predicted_bounding_boxes[indices]

        predicted_categories = predicted_categories[indices]

        return predicted_bounding_boxes, predicted_categories

Instances