forked from taesungp/swapping-autoencoder-pytorch
-
Notifications
You must be signed in to change notification settings - Fork 2
CLI interface
Abhishek Saxena edited this page Sep 26, 2021
·
1 revision
This CLI was made to test the Core python API, it's made for quick inference. There are two required arguments. The two images needed to mix.
To try it out you can run the following command after installing dependencies and weights
python -m api imgs/2.jpg imgs/1.jpg
Running the api module with -h
argument prints the help. You can see the detailed usage here.
❯ python -m api -h
usage: api [-h] [--version] [--alpha ALPHA] [--model MODEL] [--debug] Structure Style
Process some images
positional arguments:
Structure Path to Structure one
Style Path to Style two
optional arguments:
-h, --help show this help message and exit
--version show program's version number and exit
--alpha ALPHA How to mix style? '0' means 1st image, '1' means 2nd image (default: 0.75)
--model MODEL Which model to load? (default: mountain_pretrained)
--debug Print debug output? (default: False)