/pytorch_hed

The implementation of hed using pytorch.

Primary LanguagePython

pytorch_hed

You can get vgg16.pth by:

   wget https://download.pytorch.org/models/vgg16-397923af.pth

You also need make the dirs:

   mkdir checkpoints
   mkdir tmp

I have got ODS=0.771 on BSDS500 dataset with Adam.