I’m currently working on a hobby project where I’ve trained a StyleGAN on abstract art, with the output size of 1024x1024.
I’m now in the process of putting up a screen to display the generated images. My idea is to connect an edge GPU to the screen that both will run a GUI and display the images, but also run the inference. I’m then thinking about connecting a simple button to the GPIOs that triggers a new image.
This is the implementation I’ve used: https://github.com/taki0112/StyleGAN-Tensorflow
Here are the size of the checkpoint files:
Checkpoint: 2.94 GB
Index file: 75 kB
Meta file: 70.5 MB
Do you guys think a Nvidia Jetson 4GB is enought to load the model and run inference, and at the same time run a GUI and display the image on the screen?
Or would you rather go for a Jetson Xavier NX?
Or would you rather set up the Jetson Nano 4GB or Xavier NX as a “server” that only generates the images, then run a RPi or something else behind the screen that fetches the images from the Jetson?