You can check out previous work SalsaNext (GitHub), SalsaNext (Paper), and TITAN-Net (GitHub), TITAN-Net (Paper)!
The weights of the model can be found here.
This repository is for research purposes only, the use of this code is your responsibility.