Comments (6)
I ask the same question as the @Radium98
I watched the presentation on this study at the NAVER search colloquium. Your research is really impressive.
I look forward to being updated on how to learn using custom data.
from stylemapgan.
Hi guys, thank you for your interest :)
-
If you want to make your own dataset, refer to prepare_data.py.
-
I already provided LSUN models. Please refer to https://github.com/naver-ai/StyleMapGAN/blob/main/download.sh
from stylemapgan.
Hi,authors,thank you for your great work. What should I do if I want to train your model on my own dataset? I think you are supposed to tell readers how to process their own data if they want to use it for training. Another question is why not release the model about LSUN Car & Church?I want to get the model to vertify the effect your model behaved on LSUN Car & Church.
Hi, do you train the model on your own data? I trained the model on the raw data of celeba_hq (firstly,using prepare_data.py to get data.mdb ),but when I reconstruction the image,I only get the noise. Do you have any suggestion?
from stylemapgan.
Hi, do you train the model on your own data? I trained the model on the raw data of celeba_hq (firstly,using prepare_data.py to get data.mdb ),but when I reconstruction the image,I only get the noise. Do you have any suggestion?
@zackzhao
Sorry for the late reply. Was the training loss decreasing when you were training? If you only get noise, the training arguments are probably not right.
from stylemapgan.
Hi @blandocs, thanks for the amazing work. Since I have the same question and this issue still open, I decided to drop the question here.
What I wish to do with this repo is to do image reconstruction of my own image dataset, using the generate.py
with --mixing_type reconstruction
. I have tried the data preparation using prepare_data.py
with just a few images I wish to reconstruct (for testing purpose, I have only put 3 images from the dataset folder inside another folder). I have put those three images in this arrangement
(base) steven@steven-pc:~$ tree Test_20210809_StyleMapGANEncoding
Test_20210809_StyleMapGANEncoding # Parent folder
└── test # Sub folder
├── AF1500.jpg # File no. 1
├── AF1514.jpg # File no. 2
└── AF1526.jpg # File no. 3
I have no issue with dataset making. However, I got this kind of error when I using my own data.
Original Traceback (most recent call last):
File "/home/steven/anaconda3/envs/tf1py36/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 198, in _worker_loop
data = fetcher.fetch(index)
File "/home/steven/anaconda3/envs/tf1py36/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/steven/anaconda3/envs/tf1py36/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 44, in <listcomp>
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/steven/Git/StyleMapGAN/training/dataset.py", line 53, in __getitem__
img = Image.open(buffer)
File "/home/steven/anaconda3/envs/tf1py36/lib/python3.6/site-packages/PIL/Image.py", line 2959, in open
"cannot identify image file %r" % (filename if filename else fp)
PIL.UnidentifiedImageError: cannot identify image file <_io.BytesIO object at 0x7f458e00beb8>
So the question is, do I need to prepare the dataset folder exactly with same structure as CelebAHQ have?
from stylemapgan.
Hi, after trying to find out from download.sh
I can make it work with my own dataset, there might be some strict requirements about the dataset folder structure. The result looks like these (the original is on the left, and the reconstructed is on the right).
I really liking the result but I believe it can be further improved. Because I can only generate an image of size 256x256 and not 1024x1024. Was that because I used the weights from celeba_hq_5M_256_8x8.pt
? Should I use ffhq_1024_16x16.pt
instead?
I also have another question, since the output of your encoder is actually a feature map of size 64x8x8 (as mentioned in your paper), is it alright if I resize it back into one dimensional vector (of length 4096)? Will that affect the final result later? Because I need to do the a mathematical operation involving a one dimensional latent vector.
from stylemapgan.
Related Issues (20)
- Questions about metrics for reconstruction HOT 5
- Fail to open demo HOT 2
- [code]About the train.py HOT 2
- Pre-trained weights download link is invalid. HOT 6
- RuntimeError: Trying to backward through the graph a second time HOT 4
- How to edit my own images ? HOT 2
- Generated Result out of memory HOT 1
- What's the difference between normal generator and small generator? HOT 2
- Download.sh updated link points to possibly corrupted models HOT 2
- Any advice for class conditional image generation for stylemapgan HOT 2
- How to calculate or obtain GT_labels and LMDB_test_mask for other images in CelebA_HQ HOT 1
- Different sizes between downloaded lmdb data and generated data with prepare_data.py
- can i get a new pretrained network file? HOT 1
- How to do StyleMixing on custom dataset
- Encoding quality of real images HOT 3
- UnpicklingError: invalid load key, '<'. HOT 9
- what's the effect of different normalize_mode
- can't download model
- metrics in table 3
- ImportError: No module named 'upfirdn2d'
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from stylemapgan.