Potential misuse of parameter input

[ly1112]

Dear author, thank you so much for sharing your work, but I may ask if there may be a potential misuse of parameter input. Could you please take the time to check it?

in your main.py, lines 53 to 54, you use:

if (loss_type=='dilate'):
loss, loss_shape, loss_temporal = dilate_loss(target,outputs,alpha, gamma, device)
so the first arg is target(ground truth) and the second arg is output (prediction)

But in your dilate_loss.py, line 5, you use:

def dilate_loss(outputs, targets, alpha, gamma, device):

So the first arg is changed to output (prediction) and the second arg is the target(ground truth).

So may I ask should the code in main.py, lines 53 to 54, to be changed as:

if (loss_type=='dilate'):
loss, loss_shape, loss_temporal = dilate_loss(outputs,target,alpha, gamma, device)

Because in your paper, you mention the latter parameter is the ground truth.

Thanks a lot! Looking forward to your reply!