DeepSaki TensorFlow Add-On
DeepSaki is an add-on to TensorFlow. It provides a variaty of custom classes ranging from activation functions to entire models, helper functions to facilitate connectiong to your compute HW and many more! The project started as fun project to learn and to collect the code snippets I was using in my projects. Now it has been transformed into a modern SW package featuring CI/CD and a documentation.
Main-Branch:
Develop-Branch:
Some highlights:
- Layers to transform data into the frequency domain using FFTs, like FFT2D and FFT3D.
- Layers to perform calculations in the frequency domain supporting complex values like FourierPooling2D.
- Wrapper to make initializer and activation functions complex-valued.
- Utilities to auto-detect your compute hardware.
- autoencoder like models like the UNet and discriminator models like a Layout-Content-Discriminator.
- Augmentations like Cut-Mix and Cut-Out
- Custom constraints for your layers like NonNegative.
- And many more…
Installation
pip install DeepSaki
Contribute to DeepSaki
I highly encourage you to contribute to DeepSaki. Checkout our contribution guide to get started.