PavlosMelissinos

class YoloHead(Layer):

    def __init__(self, anchors, num_classes, **kwargs):
        self.anchors = anchors
        self.num_classes = num_classes
        super(YoloHead, self).__init__(**kwargs)

    def call(self, inputs, **kwargs):
        """Convert final layer features to bounding box parameters.