search

lucidrains / stylegan2-pytorch

Simplest working implementation of Stylegan2, state of the art generative adversarial network, in Pytorch. Enabling everyone to experience disentanglement
https://thispersondoesnotexist.com
MIT License
3.69k stars 585 forks source link

AssertionError: No images were found in ./data for training #208

Open gunahn opened 3 years ago

gunahn commented 3 years ago

Hi, thank you so much for making easy repo for styleGAN2.

I have a problem with using this code.

stylegan2_pytorch -- DCGAN/Knee_GAN/processed/2/test/ When I tried to use the above code, it shows error messages like this

AssertionError: No images were found in ./data for training

And DCGAN/Knee_GAN/processed/2/test/ is the path for data that I want to train. Can you help me out?

Thank you

Gun Ahn

lucidrains commented 3 years ago

@gunahn Hi! There's an error in your command; it should be stylegan2_pytorch --data DCGAN/Knee_GAN/processed/2/test/

KazuyukiFukuda commented 3 years ago

Hi @lucidrains. I've got same type of problem with @gunahn . AssertionError: No images were found in /home/nagalab/Kazuyuki/Data/train_data_resize/ for training

Here is the code that I tried, stylegan2_pytorch --data /home/nagalab/Kazuyuki/Data/train_data_resize/ --name forOriginalImg --results_dir /home/nagalab/Kazuyuki/research/stylegan2_pytorch/results --models_dir /home/nagalab/Kazuyuki/research/stylegan2_pytorch/models --batch-size 256

In /home/nagalab/Kazuyuki/Data/train_data_resize/, there are JPG images which is 1024x1024 Could you please help me?

Thank you Kazuyuki Fukuda