A pytorch reimplementation of liuwei16/CSP, their trained keras weights are loaded in pytorch. - View it on GitHub