-
Notifications
You must be signed in to change notification settings - Fork 89
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
some error use connectivity loss fuction in gpu #10
Comments
Hi @kelisiya, |
It's work , I let your backbone to DIM and use Alpha loss and Grad loss , finally I can train your model . |
@kelisiya Nice, let me know if your model achieves better results:) |
Some inference questions in Indexnet:The inference code use np.clip , but network return pred(l) isn't [0-1];
Your training loss fuction use cv.normalize and then /255 is right ?
When I don't use your pretrain and training some epoch , the loss doesn't down and the inference image have some error .
I try to use torch.clamp() in network return tensor to calculate loss fuction , or use normal lize but doesn't seem to be effective.
Do you know what causes these edges?
|
@kelisiya It is normal that the network's output is not bounded by [0, 1] due to the nature of regression. This is why postprocessing is required to eliminate unreasonale outputs. |
So are you using the cv2.normalize in your training loss fuction ? In other words , If I use sigmoid in return tensor ,it may also be effective?
|
I don't use cv.normalize. |
So only train the return tensor ,don't use any activation functions.
|
Exactly! |
Thanks for your answer.
|
other questions:
1.how to calculate alpha loss ?the input is alpha or alpha/255?2. The Indexnet input is image+trimap/255 like DIM?
|
Of course you should normalize the alpha before calculating the loss. |
Im trying to reproduce your paper . The cnn retun a tensor , but the Loss Fuction use numpy;
So I use tensor to numpy and calculate loss , the numpy to tensor . I found connectivity loss can't work.
How did you deal with it?
The text was updated successfully, but these errors were encountered: