Datasets and pretrained Models for StyleGAN3 ...

Overview

Datasets and pretrained Models for StyleGAN3 ...

Dear arfiticial friend, this is a collection of artistic datasets and models that we have put together during our ongoing stylegan3 trip at the lucid layers studios. You can use the model snapshots for instant fun, or you can work with the source datasets to train your own models.
Some models include multiple snapshots. these can give interesting variations.

tip: best viewed maximized

Updates

This Document will be updated frequently since a lot of models are still in training on higher resolutions. You find information on each update in the "releases" section. (you may "watch" this repo for getting notified on new models)


... based on Wombo Dream:

All source images in this category were generated with Wombo Dream. thanks to the cheesy api implementation of adri326 we could remotely generate thousands of images. these images were cropped to center and scaled to 1024x1024. datasets in resolutions of 256, 512 and 1024 were generated and are available for download.
Most datasets are tied to one single textpromt with some minor variations. But you may mix them together in new datasets to train multi domain models with style mixing.

Thanks to the wombo creators for their generous quota limits. since wombo is a free service, we like to share all batches that have been created.

1. Mechanical devices from the future

Dataset
Name Mechanical devices from the future
Method Wombo Dream via Wombot
Image count 2169
Dataset download 256, 512, 1024
Samples
Model: 256 px
Method stylegan3-t, Transfer learning from Landscape256
Resolution 256x256
29 kimg
Download .pkl
 network-snapshot-000029 pkl
05 kimg
Download .pkl
  network-snapshot-000005 pkl

2. Vivid Flowers

Dataset
Name Vivid Flowers
Method Wombo Dream via Wombot
Image count 793
Dataset download 256, 512, 1024
Samples
Model: 256 px
Method stylegan3-t, Transfer learning from Landscape256
Resolution 256x256
68 kimg
Download .pkl
network-snapshot-000067 pkl
12 kimg
Download .pkl
network-snapshot-000012 pkl

3. Alien with Sunglasses

Dataset
Name Alien Sunglasses
Method Wombo Dream via Wombot
Image count 1600
Dataset download 256, 512, 1024
Samples
Model: 256 px
Method stylegan3-t, Transfer learning from Landscape256
Resolution 256x256
38 kimg
Download .pkl
network-snapshot-000038 pkl

4. forest daemons

Dataset
Name forest daemons
Method Wombo Dream via Wombot
Image count 794
Dataset download 256, 512, 1024
Samples
Model: 256 px
Method stylegan3-t, Transfer learning from Landscape256
Resolution 256x256
18 kimg
Download .pkl
network-snapshot-000018 pkl
03 kimg
Download .pkl
network-snapshot-000003 pkl

5. third eye watching you

Dataset
Name third eye watching you
Method Wombo Dream via Wombot
Image count 1363
Dataset download 256, 512, 1024
Samples
Model: coming soon ...

coming soon ...

6. mars spaceport

Dataset
Name mars spaceport
Method Wombo Dream via Wombot
Image count 710
Dataset download 256, 512, 1024
Samples
Model: coming soon ...

coming soon ...

7. scifi city

Dataset
Name scifi city
Method Wombo Dream via Wombot
Image count 1245
Dataset download 256, 512, 1024
Samples
Model: 256 px
Method stylegan3-t, Transfer learning from Landscape256
Resolution 256x256
210 kimg
Download .pkl
 network-snapshot-000210 pkl
018 kimg
Download .pkl
 network-snapshot-000018 pkl
013 kimg
Download .pkl
 network-snapshot-000013 pkl
008 kimg
Download .pkl
 network-snapshot-000008 pkl

8. scifi spaceship

Dataset
Name scifi spaceship
Method Wombo Dream via Wombot
Image count 1108
Dataset download 256, 512, 1024
Samples
Model: 256
Method stylegan3-t, Transfer learning from Landscape256
Resolution 256x256
168 kimg
Download .pkl
network-snapshot-000162 pkl
128 kimg
Download .pkl
network-snapshot-000128 pkl
13 kimg
Download .pkl
network-snapshot-000013 pkl

9. yellow comic alien

Dataset
Name yellow comic alien
Method Wombo Dream via Wombot
Image count 3984
Dataset download 256, 512, 1024
Samples
Model 256x256
Method stylegan3-t, Transfer learning from Landscape256
Resolution 256x256
19 kimg
Download .pkl
network-snapshot-000019 pkl
Model 512x512
Method stylegan3-t, Transfer learning from affhq
Resolution 512x512
236 kimg
Download .pkl
network-snapshot-000236 pkl
004 kimg
Download .pkl
network-snapshot-000004 pkl

10. eternal planet earth

Dataset
Name eternal planet earth
Method Wombo Dream via Wombot
Image count 1323
Dataset download 256, 512, 1024
Samples
Model: coming soon ...

11. mechanical landscape madness

Dataset
Name mechanical landscape madness
Method Wombo Dream via Wombot
Image count 1269
Dataset download 256, 512, 1024
Samples
Model 256x256
Method stylegan3-t, Transfer learning from Landscape256
Resolution 256x256
6 kimg
Download .pkl
network-snapshot-000006 pkl
5 kimg
Download .pkl
network-snapshot-000005 pkl

12. two aliens speaking

Dataset
Name two_aliens_speaking
Method Wombo Dream via Wombot
Image count 1159
Dataset download 256, 512, 1024
Samples
Model: coming soon...

Usage

We recomend to install the official StyleGAN3 repo to your local machine. then use the "visualizer.py" to start the gui. The GUI is very comfortable to use and allows easy visual inspection of the models, in realtime. (RTX card recomended).
If you have troubles for installing on windows with anaconda, try this edited enviroment.yml file. this works with the current pytorch release (cuda11).
Alternatively you may use a colab notebook to generate images/videos from the model. Just copy the links from this page to your favorite colab notebook.
Here is a basic notebook, pre-configured for many of our models:
Open In Colab

Progress

  • prepare all datasets in resolutions 256, 512, 1024
  • create a colab notebook for model testing
  • train some datasets in 256
  • train some datasets in 512
  • train some datasets in 1024
  • create multi domain datasets and models with style mixing

Contribution

If you do continue training on a model, or train a dataset in a high resolution, it would be great to include that in this list.
(please send me a link to your .pkl file in the "issues" tab)
Also, if you made some images or videos you like to share - we would love to see your work! put everything in the issues...

License

You are welcomed to use all files for your own purposes. please include a link to this repo in your work. Thank you.
(Terms and Conditions of Wombo, Nvidia and other contributors have to be considered seperately when doing commercial projects.)

You might also like...
Reference implementation of code generation projects from Facebook AI Research. General toolkit to apply machine learning to code, from dataset creation to model training and evaluation. Comes with pretrained models.

This repository is a toolkit to do machine learning for programming languages. It implements tokenization, dataset preprocessing, model training and m

YOLOv5 🚀 is a family of object detection architectures and models pretrained on the COCO dataset
YOLOv5 🚀 is a family of object detection architectures and models pretrained on the COCO dataset

YOLOv5 🚀 is a family of object detection architectures and models pretrained on the COCO dataset, and represents Ultralytics open-source research int

Music Source Separation; Train & Eval & Inference piplines and pretrained models we used for 2021 ISMIR MDX Challenge.
Music Source Separation; Train & Eval & Inference piplines and pretrained models we used for 2021 ISMIR MDX Challenge.

Music Source Separation with Channel-wise Subband Phase Aware ResUnet (CWS-PResUNet) Introduction This repo contains the pretrained Music Source Separ

The PASS dataset: pretrained models and how to get the data -  PASS: Pictures without humAns for Self-Supervised Pretraining
The PASS dataset: pretrained models and how to get the data - PASS: Pictures without humAns for Self-Supervised Pretraining

The PASS dataset: pretrained models and how to get the data - PASS: Pictures without humAns for Self-Supervised Pretraining

Repository providing a wide range of self-supervised pretrained models for computer vision tasks.

Hierarchical Pretraining: Research Repository This is a research repository for reproducing the results from the project "Self-supervised pretraining

(ImageNet pretrained models) The official pytorch implemention of the TPAMI paper
(ImageNet pretrained models) The official pytorch implemention of the TPAMI paper "Res2Net: A New Multi-scale Backbone Architecture"

Res2Net The official pytorch implemention of the paper "Res2Net: A New Multi-scale Backbone Architecture" Our paper is accepted by IEEE Transactions o

Pretrained models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet.
Pretrained models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet.

Pretrained models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet.

Using pretrained language models for biomedical knowledge graph completion.

LMs for biomedical KG completion This repository contains code to run the experiments described in: Scientific Language Models for Biomedical Knowledg

A library for finding knowledge neurons in pretrained transformer models.
A library for finding knowledge neurons in pretrained transformer models.

knowledge-neurons An open source repository replicating the 2021 paper Knowledge Neurons in Pretrained Transformers by Dai et al., and extending the t

Comments
  • Apply the model on a interactive art demo

    Apply the model on a interactive art demo

    Thank you so much for sharing your artistic model to the community. I'm a media art student new to machine learning. I'm currently apply your 'yellow comic alien 004 kimg' model on a interactive art demo. It looks pretty cool and I'm really appriciate your time and effort! Here is a video preview: https://youtu.be/lVJnj5PLghs

    opened by HarryGuan22 2
  • transfer learning params

    transfer learning params

    Yo! Can you post what sg3 params you used for transfer learning? The results are impressive even after 005 kimg! I personally never managed to make transfer learning work, from afhqv2, maybe my datasets were too big I guess.

    opened by betterftr 0
  • Lucid Stylegan Error

    Lucid Stylegan Error

    Using colab, lucid sonic dreams, stylegan3. The name of the folder is stylegan2, but it contains stylegan3. Downloaded womboflowers pkl (256x256, 05 kimg). How to fix that issue? Is it compatible? Used it as a style, receiving following error:

    Preparing style... Loading networks from /content/drive/MyDrive/LSD/lucid-sonic-dreams/womboflowers2.pkl...

    KeyError Traceback (most recent call last) in () 37 motion_randomness = 0.8, 38 motion_harmonic = True, ---> 39 motion_percussive = False, 40 #random_seed=100 41 #class_complexity = 1

    3 frames /content/drive/MyDrive/LSD/lucid-sonic-dreams/stylegan2/torch_utils/persistence.py in _reconstruct_persistent_obj(meta) 191 192 assert meta.type == 'class' --> 193 orig_class = module.dict[meta.class_name] 194 decorator_class = persistent_class(orig_class) 195 obj = decorator_class.new(decorator_class)

    KeyError: 'FullyConnectedLayer'

    Before it "module" was used here: line 190: module = _src_to_module(meta.module_src)

    opened by Ennorath 0
Releases(v1.0.1)
Owner
lucid layers
lucid layers
Official code of the paper "Expanding Low-Density Latent Regions for Open-Set Object Detection" (CVPR 2022)

OpenDet Expanding Low-Density Latent Regions for Open-Set Object Detection (CVPR2022) Jiaming Han, Yuqiang Ren, Jian Ding, Xingjia Pan, Ke Yan, Gui-So

csuhan 64 Jan 07, 2023
Face Synthetics dataset is a collection of diverse synthetic face images with ground truth labels.

The Face Synthetics dataset Face Synthetics dataset is a collection of diverse synthetic face images with ground truth labels. It was introduced in ou

Microsoft 608 Jan 02, 2023
Pytorch Lightning Distributed Accelerators using Ray

Distributed PyTorch Lightning Training on Ray This library adds new PyTorch Lightning plugins for distributed training using the Ray distributed compu

167 Jan 02, 2023
Simple Baselines for Human Pose Estimation and Tracking

Simple Baselines for Human Pose Estimation and Tracking News Our new work High-Resolution Representations for Labeling Pixels and Regions is available

Microsoft 2.7k Jan 05, 2023
Lbl2Vec learns jointly embedded label, document and word vectors to retrieve documents with predefined topics from an unlabeled document corpus.

Lbl2Vec Lbl2Vec is an algorithm for unsupervised document classification and unsupervised document retrieval. It automatically generates jointly embed

sebis - TUM - Germany 61 Dec 20, 2022
Hub is a dataset format with a simple API for creating, storing, and collaborating on AI datasets of any size.

Hub is a dataset format with a simple API for creating, storing, and collaborating on AI datasets of any size. The hub data layout enables rapid transformations and streaming of data while training m

Activeloop 5.1k Jan 08, 2023
Neural Magic Eye: Learning to See and Understand the Scene Behind an Autostereogram, arXiv:2012.15692.

Neural Magic Eye Preprint | Project Page | Colab Runtime Official PyTorch implementation of the preprint paper "NeuralMagicEye: Learning to See and Un

Zhengxia Zou 56 Jul 15, 2022
Apollo optimizer in tensorflow

Apollo Optimizer in Tensorflow 2.x Notes: Warmup is important with Apollo optimizer, so be sure to pass in a learning rate schedule vs. a constant lea

Evan Walters 1 Nov 09, 2021
TaCL: Improving BERT Pre-training with Token-aware Contrastive Learning

TaCL: Improving BERT Pre-training with Token-aware Contrastive Learning Authors: Yixuan Su, Fangyu Liu, Zaiqiao Meng, Lei Shu, Ehsan Shareghi, and Nig

Yixuan Su 79 Nov 04, 2022
TorchFlare is a simple, beginner-friendly, and easy-to-use PyTorch Framework train your models effortlessly.

TorchFlare TorchFlare is a simple, beginner-friendly and an easy-to-use PyTorch Framework train your models without much effort. It provides an almost

Atharva Phatak 85 Dec 26, 2022
Multiview Dataset Toolkit

Multiview Dataset Toolkit Using multi-view cameras is a natural way to obtain a complete point cloud. However, there is to date only one multi-view 3D

11 Dec 22, 2022
A self-supervised learning framework for audio-visual speech

AV-HuBERT (Audio-Visual Hidden Unit BERT) Learning Audio-Visual Speech Representation by Masked Multimodal Cluster Prediction Robust Self-Supervised A

Meta Research 431 Jan 07, 2023
This repository contains numerical implementation for the paper Intertemporal Pricing under Reference Effects: Integrating Reference Effects and Consumer Heterogeneity.

This repository contains numerical implementation for the paper Intertemporal Pricing under Reference Effects: Integrating Reference Effects and Consumer Heterogeneity.

Hansheng Jiang 6 Nov 18, 2022
Fre-GAN: Adversarial Frequency-consistent Audio Synthesis

Fre-GAN Vocoder Fre-GAN: Adversarial Frequency-consistent Audio Synthesis Training: python train.py --config config.json Citation: @misc{kim2021frega

Rishikesh (ऋषिकेश) 93 Dec 17, 2022
A PyTorch Implementation of "SINE: Scalable Incomplete Network Embedding" (ICDM 2018).

Scalable Incomplete Network Embedding ⠀⠀ A PyTorch implementation of Scalable Incomplete Network Embedding (ICDM 2018). Abstract Attributed network em

Benedek Rozemberczki 69 Sep 22, 2022
PyTorch implementation of NIPS 2017 paper Dynamic Routing Between Capsules

Dynamic Routing Between Capsules - PyTorch implementation PyTorch implementation of NIPS 2017 paper Dynamic Routing Between Capsules from Sara Sabour,

Adam Bielski 475 Dec 24, 2022
This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Semantic Segmentation.

Swin Transformer for Semantic Segmentation of satellite images This repo contains the supported code and configuration files to reproduce semantic seg

23 Oct 10, 2022
CRNN With PyTorch

CRNN-PyTorch Implementation of https://arxiv.org/abs/1507.05717

Vadim 4 Sep 01, 2022
PyTorch implementation of the Crafting Better Contrastive Views for Siamese Representation Learning

Crafting Better Contrastive Views for Siamese Representation Learning This is the official PyTorch implementation of the ContrastiveCrop paper: @artic

249 Dec 28, 2022
Solutions of Reinforcement Learning 2nd Edition

Solutions of Reinforcement Learning, An Introduction

YIFAN WANG 1.4k Dec 30, 2022