overview

This is the PyTorch implementation of XXX

Requirements

• torch==1.3.0+cpu
• scipy==1.3.1
• numpy==1.17.3

pip install -r requirements.txt

Preparation

• Dataset download link: Baidu Netdisk; Google Drive
• Checkpoint download link: Baidu Netdisk; Google Drive
• Tree Arrangement

.
├── checkpoint
│   ├── model
│   └── pretrained_model
│       ├── with_10BS_model.pth
│       ├── with_15BS_model.pth
│       ├── with_20BS_model.pth
│       ├── with_25BS_model.pth
│       └── with_5BS_model.pth
├── dataset
│   ├── test_5.mat
│   ├── train_5.mat
│   └── ...
├── dataset.py
├── loss.py
├── network.py
├── requirements.txt
├── README.md
├── test.py
├── train.py
└── utils.py

Run

git clone https://github.com/IIT-Lab/MBPANet.git
cd MBPANet
python train.py
python test.py