### Usage:

---

To train a model, run `Train.py`.  You can modify training settings

To test a model, run `Test.py`. 

### Saved Models:

---

The trained model files for FFSP are too large to be included in the folder, so we provide download links below instead. Please place the models for each size in the `FFSP_MatNet/result/saved_FFSP` folder.

* [saved_FFSP20_model](https://drive.google.com/drive/folders/1YQpI2NI7h7WW3AHnmUTga6xJ9wNbD1pr?usp=sharing)
* [saved_FFSP50_model](https://drive.google.com/drive/folders/1lDqL0rWIzhUbGvgZh6HsjFijOE9zZOqI?usp=sharing)
* [saved_FFSP100_model](https://drive.google.com/drive/folders/1ScbnJms5LITaMu4uSfbB5kkzbpkUo5AF?usp=sharing)

You can find the initial checkpoint used for training in the original [MatNet](https://github.com/yd-kwon/MatNet) code repository. Please place the models for each size in the `FFSP_MatNet/result/init_FFSP` folder.

### Datasets:

---

The test dataset files for FFSP are too large to be included in the folder, so we provide download links below instead. Please place the datasets in the `data` folder.

* [Test datasets](https://drive.google.com/drive/folders/16KcadIHDxQqRdZkLXvxk8PhTARUnJOiq?usp=sharing)

