def forward(self, input, target):
    input = input.squeeze(1)
    target = target.squeeze(1)
    loss = lovasz_hinge(input, target, per_image=True)
    return loss
def forward(self, input, target):
    input = input.squeeze(1)
    target = target.squeeze(1)
    loss = lovasz_hinge(input, target, per_image=True)
    return loss