buntyke/pytorch-hed

Does anyone get good results for training?

ayst123 opened this issue · 7 comments

ayst123 commented

I follow the train.py to train the model with the default hyper parameters in the code. However, the results are not good. Does anyone successfully train the model yet? Appreciate if providing the hyper parameters. Thanks

weixia1 commented

the same to me. can anyone offer a good trained model? Thanks

xjock commented

No, the results are so bad.

shubhamag commented

Did anyone manage to resolve this ? Or if a pretrained pytorch model is available ?

xwjabc commented

Hi guys, I reimplemented a PyTorch version of HED which can get ODS=0.787 OIS=0.807: https://github.com/xwjabc/hed
Hope it can be somewhat helpful.

  • 👍3
  • 😄1
  • ❤️1
zxy14120448 commented

The test result is so so so bad!!!

xwjabc commented

The test result is so so so bad!!!

Perhaps you can give a try for my implementation :)

herleeyandi commented

Here are some results of this repository for the curious one. I am using the max epoch=15 and train about 5 hours.
image