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
Alias-Free Generative Adversarial Networks (StyleGAN3) Official PyTorch implementation

Alias-Free Generative Adversarial Networks (StyleGAN3) Official PyTorch implementation

NVIDIA Research Projects 4.8k Jan 9, 2023
Facebook Research 605 Jan 2, 2023
Pretrained SOTA Deep Learning models, callbacks and more for research and production with PyTorch Lightning and PyTorch

Pretrained SOTA Deep Learning models, callbacks and more for research and production with PyTorch Lightning and PyTorch

Pytorch Lightning 1.4k Jan 1, 2023
An easy way to build PyTorch datasets. Modularly build datasets and automatically cache processed results

EasyDatas An easy way to build PyTorch datasets. Modularly build datasets and automatically cache processed results Installation pip install git+https

Ximing Yang 4 Dec 14, 2021
Deep Learning Datasets Maker is a QGIS plugin to make datasets creation easier for raster and vector data.

Deep Learning Dataset Maker Deep Learning Datasets Maker is a QGIS plugin to make datasets creation easier for raster and vector data. How to use Down

deepbands 25 Dec 15, 2022
Cl datasets - PyTorch image dataloaders and utility functions to load datasets for supervised continual learning

Continual learning datasets Introduction This repository contains PyTorch image

berjaoui 5 Aug 28, 2022
Pretrained Pytorch face detection (MTCNN) and recognition (InceptionResnet) models

Face Recognition Using Pytorch Python 3.7 3.6 3.5 Status This is a repository for Inception Resnet (V1) models in pytorch, pretrained on VGGFace2 and

Tim Esler 3.3k Jan 4, 2023
This project provides an unsupervised framework for mining and tagging quality phrases on text corpora with pretrained language models (KDD'21).

UCPhrase: Unsupervised Context-aware Quality Phrase Tagging To appear on KDD'21...[pdf] This project provides an unsupervised framework for mining and

Xiaotao Gu 146 Dec 22, 2022
Official PyTorch implementation and pretrained models of the paper Self-Supervised Classification Network

Self-Classifier: Self-Supervised Classification Network Official PyTorch implementation and pretrained models of the paper Self-Supervised Classificat

Elad Amrani 24 Dec 21, 2022
Measuring and Improving Consistency in Pretrained Language Models

ParaRel ?? This repository contains the code and data for the paper: Measuring and Improving Consistency in Pretrained Language Models as well as the

Yanai Elazar 26 Dec 2, 2022