search

davidbau / rewriting

Rewriting a Deep Generative Model, ECCV 2020 (oral). Interactive tool to directly edit the rules of a GAN to synthesize scenes with objects added, removed, or altered. Change StyleGANv2 to make extravagant eyebrows, or horses wearing hats.
https://rewriting.csail.mit.edu/
MIT License
537 stars 76 forks source link

How to rewrite on other dataset #4

Open dayby opened 3 years ago

dayby commented 3 years ago

sorry to disturb you. As a beginner, I am very interested in your work, and I would like to ask how to rewrite on other dataset

davidbau commented 3 years ago

Two steps:

(1) The first step is to train a styleganv2 on your dataset. This is most of the work. You can Google to find various tutorials to do this e.g., from very brief youtube tutorials https://www.youtube.com/watch?v=OVY-CQD0nJk to very involved explanation guides https://www.gwern.net/Faces#stylegan-2.

(2) Once you're done, the second step is to convert your trained styleganv2 to pytorch using the steps at the bottom of the readme in this repository. See the instructions here: https://github.com/davidbau/rewriting#rewriting-your-own-models