hed
hed copied to clipboard
Question about outputting 5 side-outputs
Hello, thanks for sharing! I want to get the five side-outputs however Pytorch is different to caffe. Is there any suggestions? Thanks a lot! Looking forward to your reply!
In the training stage, the code will output all five side outputs together as a png file. You may refer to https://github.com/xwjabc/hed/blob/master/hed.py#L250 in the code.