Sign inSign up

heatonresearch/stylegan3

By heatonresearch

•Updated over 3 years ago

Docker image to run/train StyleGAN3 images.

Image
0

378

heatonresearch/stylegan3 repository overview

⁠Docker file to run StyleGAN3, docker-stylegan3

Jeff Heaton's⁠ Docker image for running Stylegan3 with GPU. This dockerfile is intended to be run from Docker. To start a container from this image run something similar to the following:

sudo docker run -it --gpus all -u $(id -u):$(id -g) -p 8888:8888  -v /home/username/:/content/mnt heatonresearch/stylegan3ash

The above command establishes the following:

  • -u $(id -u):$(id -g) - This causes the user to be logged into Docker to be the same as the user running the Docker command. This ensures that all permissions and ownerships are correct on the mounted volumes.
  • /home/username/:/content/mnt - You should mount a volume to access your images and store results to.
  • /bin/bash - Optional: You can start the BASH shell to allow you to execute commands.
cd /home/stylegan2-ada/
python dataset_tool.py create_from_images /mnt/datasets/fish /mnt/data/fish

Refer to readme.ipynb for information on training and converting images.

Tag summary

Content type

Image

Digest

sha256:ab64fe836…

Size

6.2 GB

Last updated

over 3 years ago

docker pull heatonresearch/stylegan3