diff --git a/docs/en_US/tutorials/psgan.md b/docs/en_US/tutorials/psgan.md index 278ee451f811674be23a5442edd86af9d958d298..74af4a6bb413b4aaf03c63c0ddaf4577142d8cff 100644 --- a/docs/en_US/tutorials/psgan.md +++ b/docs/en_US/tutorials/psgan.md @@ -19,7 +19,7 @@ python tools/psgan_infer.py \ --config-file configs/makeup.yaml \ --model_path /your/model/path \ --source_path docs/imgs/ps_source.png \ - --reference_dir docs/imgs/ref/ps_ref \ + --reference_dir docs/imgs/ref \ --evaluate-only True ``` **params:** diff --git a/docs/zh_CN/tutorials/psgan.md b/docs/zh_CN/tutorials/psgan.md index 1f33638d020a4b070ff40cf82c143efad34026ce..5d55f146dc46ab9f6df9bb4cda3a4d722efe3aed 100644 --- a/docs/zh_CN/tutorials/psgan.md +++ b/docs/zh_CN/tutorials/psgan.md @@ -19,7 +19,7 @@ python tools/psgan_infer.py \ --config-file configs/makeup.yaml \ --model_path /your/model/path \ --source_path docs/imgs/ps_source.png \ - --reference_dir docs/imgs/ref/ps_ref \ + --reference_dir docs/imgs/ref \ --evaluate-only True ``` **参数说明:** diff --git a/ppgan/apps/psgan_predictor.py b/ppgan/apps/psgan_predictor.py index fef0f9140928e571dc4c1614b676f9283dda1445..488a7a8b61b206dc8c9a6df9ee1bf74ddc3a7729 100644 --- a/ppgan/apps/psgan_predictor.py +++ b/ppgan/apps/psgan_predictor.py @@ -210,8 +210,10 @@ class PSGANPredictor(BasePredictor): ref_img_name = os.path.split(reference_path)[1] if not os.path.exists(self.output_path): - os.makedirs(sefl.output_path) + os.makedirs(self.output_path) save_path = os.path.join(self.output_path, 'transfered_ref_' + ref_img_name) image.save(save_path) + print('Transfered image {} has been saved!'.format(save_path)) + print('done!!!')