Generation of dataset.json for the pytorch version

NVlabs / stylegan2-ada-pytorch

StyleGAN2-ADA - Official PyTorch implementation

https://arxiv.org/abs/2006.06676

Other

4.13k stars 1.16k forks source link

Generation of dataset.json for the pytorch version #53

Open ahripanto opened 3 years ago

ahripanto commented 3 years ago

How I can generate dataset.json for training a blank new model? My Datasets are currently sort in a folder structure. I want to train on Full body pictures. I see in some datasets, the area of the faces are marked in this file. If exists a guide or documentation how to create the dataset meta file and what arguments are in use? Thanks

nurpax commented 3 years ago

Try if

python dataset_tool.py —help

answers your question. I added a section about dataset.json a while ago.

The dataset.json file is currently used only for class labels.

ahripanto commented 3 years ago

I see and read the help file, but how exactly you mean class labels? Can a image have multiple classes? And need I to write the json with nano by me own or exists some tools for this? I know the tool LabelImg. Can i use this tool for my task? Thanks.

dokluch commented 3 years ago

I see and read the help file, but how exactly you mean class labels? Can a image have multiple classes? And need I to write the json with nano by me own or exists some tools for this? I know the tool LabelImg. Can i use this tool for my task? Thanks.

From my experience you may generate you own json file as described in help and it works just fine.