Question about outputting 5 side-outputs

xwjabc / hed

A PyTorch reimplementation of Holistically-Nested Edge Detection

169 stars 42 forks source link

Question about outputting 5 side-outputs #11

Open Johnhave opened 5 years ago

Johnhave commented 5 years ago

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!

xwjabc commented 5 years ago

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.