Hi r/opensource!
I’m excited to share with you all my first open-source project, NectarGAN!
https://github.com/ZacharyBork/NectarGAN/
NectarGAN is comprised of two main components:
A modular PyTorch-based API for building, training, and testing cGAN models. The NectarGAN API includes drop-in components for managing and tracking training configurations and experiment data, handling and logging loss functions during training, building and applying complex schedules for losses and learning rates, and much more. With it, you can quickly take models from concept to deployment with minimal boilerplate code. 1.
The NectarGAN Toolbox, a PySide6-based graphical dashboard for assembling, train…
Hi r/opensource!
I’m excited to share with you all my first open-source project, NectarGAN!
https://github.com/ZacharyBork/NectarGAN/
NectarGAN is comprised of two main components:
A modular PyTorch-based API for building, training, and testing cGAN models. The NectarGAN API includes drop-in components for managing and tracking training configurations and experiment data, handling and logging loss functions during training, building and applying complex schedules for losses and learning rates, and much more. With it, you can quickly take models from concept to deployment with minimal boilerplate code. 1.
The NectarGAN Toolbox, a PySide6-based graphical dashboard for assembling, training, and testing models, reviewing experiment results, processing datasets, converting models to ONNX for deployment, and testing your converted models. You can oversee the entire lifecycle of your model from end to end without ever leaving the interface or writing a line of code.
NectarGAN also includes a Docker build setup and a dedicated CLI wrapper for the container. This allows you to train and test models in a containerized environment, with live file IO to the host machine, using Visdom for real-time data visualization during training.
NectarGAN has been tested on Windows and Linux (Debian/Ubuntu), and is available under the Apache 2.0 license.
A little bit about me:
I’m a CG pipeline TD/Tech Artist, and a while back I got really in to the idea of using machine learning models to generate textures for 3D models in Houdini. That led to me wanting to learn more about how the models work, which led to me wanting to build one, which led to NectarGAN. I’ve never actually released a piece of open-source software before, so I’ve been a tiny bit nervous putting it out there. This has been a passion project of mine for a while now, though, so I’m super excited to share it.
Any and all feedback is appreciated! If you’re interested in contributing, there is a contribution guide in the repository. If you have any questions, please feel free to ask! I hope you all like it!