3

我正在尝试实现 StyleGAN2-ADA PyTorch:https ://github.com/NVlabs/stylegan2-ada-pytorch 。

在 GitHub 存储库中,它声明了以下内容:

上面的代码需要通过 PYTHONPATH 访问 torch_utils 和 dnnlib。它不需要网络本身的源代码——它们的类定义是通过 torch_utils.persistence 从 pickle 加载的。

这是什么意思,我该怎么做?

4

1 回答 1

1

假设您已克隆了此 repo 的源代码,/somepath/stylegan2-ada-pytorch这意味着您引用的目录分别位于/somepath/stylegan2-ada-pytorch/torch_utils/somepath/stylegan2-ada-pytorch/dnnlib

现在假设您有一个要访问此代码的 python 脚本。它可以在你机器上的任何地方,只要你把它添加到你的 python 脚本的顶部:

import os
import sys

#save the literal filepath to both directories as strings
tu_path = os.path.join('somepath','stylegan2-ada-pytorch','torch_utils')
dnnlib_path = os.path.join('somepath','stylegan2-ada-pytorch','dnnlib')

#add those strings to python path
sys.path.append(tu_path)
sys.path.append(dnnlib_path )

请注意,这只在 python 脚本运行期间将这些位置添加到 PYTHONPATH,因此您需要在任何打算使用这些库的 python 脚本的顶部使用它。

于 2021-09-14T01:36:13.677 回答