pytorch_spn
To build, install pytorch and run:
$ sh make.sh
See left_right_demo.py for usage:
$ mv left_right_demo.py ../
$ python left_right_demo.py
The original codes (caffe) and models will be relesed HERE.
Extension package for spatial propagation network in pytorch.
http://research.nvidia.com/publication/2017-12_Learning-Affinity-via
To build, install pytorch and run:
$ sh make.sh
See left_right_demo.py for usage:
$ mv left_right_demo.py ../
$ python left_right_demo.py
The original codes (caffe) and models will be relesed HERE.
Extension package for spatial propagation network in pytorch.
http://research.nvidia.com/publication/2017-12_Learning-Affinity-via