Please use the prereq.sh
script to install the required packages.
Please download required datasets and specify their paths in corresponding Python files.
Please modify the train_classifer.py
file for training the classifier. Two options are available: --mode train
and --mode test
Please modify the train_vae.py
file for training the VAE.
Please modify the train_halftone.py
file for training the adapters and the prompt. For training arguments, please refer to the useful.sh
script.
Use test_adapter.py
to do inference using list of halftone images in a txt file. Please refer to the useful.sh
script for the arguments.
Use valid_halftone.py
to get the PSNR and SSIM scores for a test dataset.
Use valid_halftone_fid.py
to get the FID scores for a test dataset.
Models shall be stored in the models
folder. Please refer to the README.md
script in that folder for more information.