previtus/demo-gan-explorer
check https://github.com/previtus/GAN_explorer
Image
0
3.3K
previtus/demo-gan-explorer repository overview
Docker demo for GAN Explorer - https://github.com/previtus/GAN_explorer
0 Setup
Please install docker and the nvidia-docker following instructions at https://github.com/NVIDIA/nvidia-docker
1 Demo with default network for GAN Explorer:
Default run with a pretrained model (showing the karras2018iclr-lsun-car-256x256.pkl model):
sudo docker run -it --rm --user=$(id -u $USER):$(id -g $USER) --env="DISPLAY" --volume="/tmp/.X11-unix:/tmp/.X11-unix:rw" --env="QT_X11_NO_MITSHM=1" -v $(pwd)/renders/:/home/GAN_explorer/renders/ --gpus all -ti previtus/demo-gan-explorer
For a slightly more involved demo you can try:
sudo docker run -it --rm --user=$(id -u $USER):$(id -g $USER) --env="DISPLAY" --volume="/tmp/.X11-unix:/tmp/.X11-unix:rw" --env="QT_X11_NO_MITSHM=1" -v $(pwd)/renders/:/home/GAN_explorer/renders/ --gpus all -ti previtus/demo-gan-explorer bash
Followed by:
cd GAN_explorer
python demo.py -network karras2018iclr-lsun-car-256x256.pkl
2 Demo with custom network for GAN Explorer:
Place your network net.pkl into networks/ and then start the Docker with:
sudo docker run -it --rm --user=$(id -u $USER):$(id -g $USER) --env="DISPLAY" --volume="/tmp/.X11-unix:/tmp/.X11-unix:rw" --env="QT_X11_NO_MITSHM=1" -v $(pwd)/renders/:/home/GAN_explorer/renders/ -v $(pwd)/networks/:/home/GAN_explorer/networks/ --gpus all -ti previtus/demo-gan-explorer
Acknowledgements
This docker is using the trained Progressive Growing GAN models from tkarras/progressive_growing_of_gans.
Tag summary
Content type
Image
Digest
Size
2.1 GB
Last updated
over 6 years ago
docker pull previtus/demo-gan-explorer