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
"Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devices", official implementation

Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devices This repository contains the official PyTorch implemen

Yandex Research 21 Oct 18, 2022
Code to reproduce the results for Compositional Attention

Compositional-Attention This repository contains the official implementation for the paper Compositional Attention: Disentangling Search and Retrieval

Sarthak Mittal 58 Nov 30, 2022
Official repository for: Continuous Control With Ensemble DeepDeterministic Policy Gradients

Continuous Control With Ensemble Deep Deterministic Policy Gradients This repository is the official implementation of Continuous Control With Ensembl

4 Dec 06, 2021
git《FSCE: Few-Shot Object Detection via Contrastive Proposal Encoding》(CVPR 2021) GitHub: [fig8]

FSCE: Few-Shot Object Detection via Contrastive Proposal Encoding (CVPR 2021) This repo contains the implementation of our state-of-the-art fewshot ob

233 Dec 29, 2022
Tilted Empirical Risk Minimization (ICLR '21)

Tilted Empirical Risk Minimization This repository contains the implementation for the paper Tilted Empirical Risk Minimization ICLR 2021 Empirical ri

Tian Li 40 Nov 28, 2022
Implementation of ConvMixer for "Patches Are All You Need? 🤷"

Patches Are All You Need? 🤷 This repository contains an implementation of ConvMixer for the ICLR 2022 submission "Patches Are All You Need?" by Asher

CMU Locus Lab 934 Jan 08, 2023
Robbing the FED: Directly Obtaining Private Data in Federated Learning with Modified Models

Robbing the FED: Directly Obtaining Private Data in Federated Learning with Modified Models This repo contains a barebones implementation for the atta

16 Dec 04, 2022
Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch

Rotary Embeddings - Pytorch A standalone library for adding rotary embeddings to transformers in Pytorch, following its success as relative positional

Phil Wang 110 Dec 30, 2022
Implementation of Perceiver, General Perception with Iterative Attention in TensorFlow

Perceiver This Python package implements Perceiver: General Perception with Iterative Attention by Andrew Jaegle in TensorFlow. This model builds on t

Rishit Dagli 84 Oct 15, 2022
BrainGNN - A deep learning model for data-driven discovery of functional connectivity

A deep learning model for data-driven discovery of functional connectivity https://doi.org/10.3390/a14030075 Usman Mahmood, Zengin Fu, Vince D. Calhou

Usman Mahmood 3 Aug 28, 2022
Official code for 'Pixel-wise Energy-biased Abstention Learning for Anomaly Segmentationon Complex Urban Driving Scenes'

PEBAL This repo contains the Pytorch implementation of our paper: Pixel-wise Energy-biased Abstention Learning for Anomaly Segmentation on Complex Urb

Yu Tian 117 Jan 03, 2023
Visual Adversarial Imitation Learning using Variational Models (VMAIL)

Visual Adversarial Imitation Learning using Variational Models (VMAIL) This is the official implementation of the NeurIPS 2021 paper. Project website

14 Nov 18, 2022
An open-source Deep Learning Engine for Healthcare that aims to treat & prevent major diseases

AlphaCare Background AlphaCare is a work-in-progress, open-source Deep Learning Engine for Healthcare that aims to treat and prevent major diseases. T

Siraj Raval 44 Nov 05, 2022
[ICCV2021] Learning to Track Objects from Unlabeled Videos

Unsupervised Single Object Tracking (USOT) 🌿 Learning to Track Objects from Unlabeled Videos Jilai Zheng, Chao Ma, Houwen Peng and Xiaokang Yang 2021

53 Dec 28, 2022
This is the repository of shape matching algorithm Iterative Rotations and Assignments (IRA)

Description This is the repository of shape matching algorithm Iterative Rotations and Assignments (IRA), described in the publication [1]. Directory

MAMMASMIAS Consortium 6 Nov 14, 2022
Attention-based CNN-LSTM and XGBoost hybrid model for stock prediction

Attention-based CNN-LSTM and XGBoost hybrid model for stock prediction Requirements The code has been tested running under Python 3.7.4, with the foll

zshicode 84 Jan 01, 2023
Recovering Brain Structure Network Using Functional Connectivity

Recovering-Brain-Structure-Network-Using-Functional-Connectivity Framework: Papers: This repository provides a PyTorch implementation of the models ad

5 Nov 30, 2022
Integrated physics-based and ligand-based modeling.

ComBind ComBind integrates data-driven modeling and physics-based docking for improved binding pose prediction and binding affinity prediction. Given

Dror Lab 44 Oct 26, 2022
Learning to Identify Top Elo Ratings with A Dueling Bandits Approach

Learning to Identify Top Elo Ratings We propose two algorithms MaxIn-Elo and MaxIn-mElo to solve the top players identification on the transitive and

2 Jan 14, 2022
本步态识别系统主要基于GaitSet模型进行实现

本步态识别系统主要基于GaitSet模型进行实现。在尝试部署本系统之前,建立理解GaitSet模型的网络结构、训练和推理方法。 系统的实现效果如视频所示: 演示视频 由于模型较大,部分模型文件存储在百度云盘。 链接提取码:33mb 具体部署过程 1.下载代码 2.安装requirements.txt

16 Oct 22, 2022