This is Pytorch implementation of our TCYB(IEEE TRANSACTIONS ON CYBERNETICS) paper.
Download the following pre-trained models BaiduYun Extracted code: wbpw into CANet folder.
Download the our prediction result maps BaiduYun Extracted code: wbpw into CANet folder.
- Set the main.py and CANet.py path.
- Set the mode parameters of main.py is train.
- Set the datasets path in main.py (Actually dataname of main.py).
If you have some question: cxpanzefeng@163.com