Saak Transform
This is a reimplementation of the paper On Data-Driven Saak Transform (https://arxiv.org/abs/1710.04176), maintained by Jiali Duan and Yueru Chen.
Table of Content
-
[Dataset] ( Hand-written digits classification)
- [Mnist] ( train set: 60000, 28x28. We used the same with downloaded from http://yann.lecun.com/exdb/mnist/)
-
[Installation] (sklearn and Pytorch)
- [Sklearn Installation] Refer to http://scikit-learn.org/stable/install.html)
- [Pytorch Installation] (Refer to http://pytorch.org)
- [Optional: Jupyter Notebook] (Refer to http://jupyter.org/install.html)
-
[How to] (Forward and Inverse Transform)
- Forward Transform:
multi_stage_saak_trans
- Inverse Transform:
toy_recon(outputs,filters)
- Forward Transform:
-
[To-do list]
- One-stage Saak Transform
- Multi-stage Saak Transform
- Inverse Transform
-
[Other Code]
- [notebook] multi-stage_saak.ipynb
- [dataset I/O] datasets.py, utils.py
Contact me
Jiali Duan (Email: jli.duan@gmail.com)
Yueru Chen (Email: yueruche@usc.edu)