💾
SynZ Dataset in Kaggle
💾
UISketch Dataset in Kaggle
- Follow the steps and run the script from the
notebooks
folder- Notebook
RICO_data_extraction.ipynb
converts RICO annotations to a CSV file - Notebook
RICO_to_SynZ_annotations.ipynb
converts RICO annotations to SynZ annotations for further steps
- Notebook
To generate SynZ sketches from scratch,
-
Download all the UISketch dataset to the
data
folder- UISketch dataset from kaggle
-
Either run the notebooks described in previous section to generate
SynZ_ready_annotations.csv
file or download the SynZ annotations from here -
Install the necessary requirements from
requirements.txt
filepip install -r requirements.txt
-
Run the following command to generate SynZ dataset
python generate_synz_dataset.py \ --annotation_csv_path=./data/SynZ_ready_annotations.csv \ --uisketch_labels_csv_path=./data/uisketch/labels.csv \ --output_path=./data/SynZ \ --number_of_versions=1 \ run
- Download the trained SynZ checkpoint from here to
models
folder - Extract the
synz_checkpoint.tar.gz
folder to acquire the MetaMorph model's trained weights frommodel_final.pth
👤 Vinoth Pandian
- Website: vinoth.info
- Github: @vinothpandian
- LinkedIn: @vinothpandian
👤 Sarah Suleri
- Website: sarahsuleri.info
- LinkedIn: @sarahsuleri