Docker image to run/train StyleGAN3 images.
378
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.
Content type
Image
Digest
sha256:ab64fe836…
Size
6.2 GB
Last updated
over 3 years ago
docker pull heatonresearch/stylegan3