imlixinyang / HiSD

Official pytorch implementation of paper "Image-to-image Translation via Hierarchical Style Disentanglement" (CVPR 2021 Oral).

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

test代码有一些错误

jweihe opened this issue · comments

有一些错误 在test.py中 os.makedirs(opts.output_dir, exist_ok=True)应改为os.makedirs(opts.output_path, exist_ok=True) 并且后面的局部变量也有错误

@hithejunwei 不好意思,test文件可能在我清理的时候因为没有经过验证有一些错误,我会尽快更正,在这之前你可以使用easy_use文件进行测试,这两个文件已经经过了验证。

不客气 我在里面目前发现两处错误 改正之后代码运行是没有问题的
第二处是vutils.save_image(((x_trg + 1)/ 2).data, os.path.join(opts.output_path, f'{os.path.basename(input)}output{i}.jpg'), padding=0) 这里i没有定义

@hithejunwei 我已经进行了更正 谢谢你纠正 有其他问题随时提issue~