How can I use the pre-trained models on my own images? about stargan

[gitama2016]

did you solve this ???

[AidasK]

python main.py --mode test --dataset RaFD --rafd_crop_size 256 --image_size 256 --c_dim 5 --rafd_image_dir ../MYIMAGES/ --sample_dir stargan_custom/samples --log_dir stargan_custom/logs                  --model_save_dir stargan_celeba_256/models --result_dir stargan_custom/results

MYIMAGES folder has to have subfolders in it. Though results with pretrained model are no good, maybe I missed some option.

[Tony607]

To have better result, first create a custom list_attr_celeba_custom.txt in folder data/ with images listings, remember to have your custom images in data/custom/images to match with the names. Then run.

python main.py --mode test --dataset CelebA --image_size 256 --selected_attrs Black_Hair Blond_Hair Brown_Hair Male Young --c_dim 5 --celeba_image_dir data/custom/images --attr_path data/list_attr_celeba_custom.txt --model_save_dir stargan_celeba_256/models --result_dir stargan_celeba_256_custom/results

[yunjey]

@AidasK Cropping input images properly will improve the visual quality of translation results. (e.g., removing black background and resizing them to 256x256)

[anandbhattad]

@Tony607 What do you mean by image listings?