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
Image Restoration Using Swin Transformer for VapourSynth

SwinIR SwinIR function for VapourSynth, based on https://github.com/JingyunLiang/SwinIR. Dependencies NumPy PyTorch, preferably with CUDA. Note that t

Holy Wu 11 Jun 19, 2022
Weakly Supervised End-to-End Learning (NeurIPS 2021)

WeaSEL: Weakly Supervised End-to-end Learning This is a PyTorch-Lightning-based framework, based on our End-to-End Weak Supervision paper (NeurIPS 202

Auton Lab, Carnegie Mellon University 131 Jan 06, 2023
Code for the paper "Next Generation Reservoir Computing"

Next Generation Reservoir Computing This is the code for the results and figures in our paper "Next Generation Reservoir Computing". They are written

OSU QuantInfo Lab 105 Dec 20, 2022
KITTI-360 Annotation Tool is a framework that developed based on python(cherrypy + jinja2 + sqlite3) as the server end and javascript + WebGL as the front end.

KITTI-360 Annotation Tool is a framework that developed based on python(cherrypy + jinja2 + sqlite3) as the server end and javascript + WebGL as the front end.

86 Dec 12, 2022
High frequency AI based algorithmic trading module.

Flow Flow is a high frequency algorithmic trading module that uses machine learning to self regulate and self optimize for maximum return. The current

59 Dec 14, 2022
PyArmadillo: an alternative approach to linear algebra in Python

PyArmadillo is a linear algebra library for the Python language, with an emphasis on ease of use.

Terry Zhuo 58 Oct 11, 2022
Semi-Supervised Signed Clustering Graph Neural Network (and Implementation of Some Spectral Methods)

SSSNET SSSNET: Semi-Supervised Signed Network Clustering For details, please read our paper. Environment Setup Overview The project has been tested on

Yixuan He 9 Nov 24, 2022
Chainer Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)

fcn - Fully Convolutional Networks Chainer implementation of Fully Convolutional Networks. Installation pip install fcn Inference Inference is done as

Kentaro Wada 218 Oct 27, 2022
U-Net: Convolutional Networks for Biomedical Image Segmentation

Deep Learning Tutorial for Kaggle Ultrasound Nerve Segmentation competition, using Keras This tutorial shows how to use Keras library to build deep ne

Yihui He 401 Nov 21, 2022
Azion the best solution of Edge Computing in the world.

Azion Edge Function docker action Create or update an Edge Functions on Azion Edge Nodes. The domain name is the key for decision to a create or updat

8 Jul 16, 2022
RipsNet: a general architecture for fast and robust estimation of the persistent homology of point clouds

RipsNet: a general architecture for fast and robust estimation of the persistent homology of point clouds This repository contains the code asscoiated

Felix Hensel 14 Dec 12, 2022
A simple implementation of Kalman filter in Multi Object Tracking

kalman Filter in Multi-object Tracking A simple implementation of Kalman filter in Multi Object Tracking 本实现是在https://github.com/liuchangji/kalman-fil

124 Dec 29, 2022
A Python library for Deep Probabilistic Modeling

Abstract DeeProb-kit is a Python library that implements deep probabilistic models such as various kinds of Sum-Product Networks, Normalizing Flows an

DeeProb-org 46 Dec 26, 2022
The official implementation of A Unified Game-Theoretic Interpretation of Adversarial Robustness.

This repository is the official implementation of A Unified Game-Theoretic Interpretation of Adversarial Robustness. Requirements pip install -r requi

Jie Ren 17 Dec 12, 2022
Source code of CIKM2021 Long Paper "PSSL: Self-supervised Learning for Personalized Search with Contrastive Sampling".

PSSL Source code of CIKM2021 Long Paper "PSSL: Self-supervised Learning for Personalized Search with Contrastive Sampling". It consists of the pre-tra

2 Dec 21, 2021
FewBit — a library for memory efficient training of large neural networks

FewBit FewBit — a library for memory efficient training of large neural networks. Its efficiency originates from storage optimizations applied to back

24 Oct 22, 2022
A supplementary code for Editable Neural Networks, an ICLR 2020 submission.

Editable neural networks A supplementary code for Editable Neural Networks, an ICLR 2020 submission by Anton Sinitsin, Vsevolod Plokhotnyuk, Dmitry Py

Anton Sinitsin 32 Nov 29, 2022
This repository contains a CBIR system that uses swin transformer to extract image's feature.

Swin-transformer based CBIR This repository contains a CBIR(content-based image retrieval) system. Here we use Swin-transformer to extract query image

JsHou 12 Nov 17, 2022
A Learning-based Camera Calibration Toolbox

Learning-based Camera Calibration A Learning-based Camera Calibration Toolbox Paper The pdf file can be found here. @misc{zhang2022learningbased,

Eason 14 Dec 21, 2022
CoTr: Efficiently Bridging CNN and Transformer for 3D Medical Image Segmentation

CoTr: Efficient 3D Medical Image Segmentation by bridging CNN and Transformer This is the official pytorch implementation of the CoTr: Paper: CoTr: Ef

218 Dec 25, 2022