未验证 提交 d3f87261 编写于 作者: L lijianshe02 提交者: GitHub

fix psgan doc (#92)

上级 9fcfc6e1
......@@ -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:**
......
......@@ -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
```
**参数说明:**
......
......@@ -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!!!')
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册