我在运行stylegan3的时候遇到了这个错误,我试过很多方法都没弄成
Setting up PyTorch plugin "bias_act_plugin"... Failed!
Traceback (most recent call last):
File "E:\GAN\stylegan3-main\train.py", line 286, in <module>
main() # pylint: disable=no-value-for-parameter
File "C:\Users\28636\AppData\Local\Programs\Python\Python39\lib\site-packages\click\core.py", line 1157, in __call__
return self.main(*args, **kwargs)
File "C:\Users\28636\AppData\Local\Programs\Python\Python39\lib\site-packages\click\core.py", line 1078, in main
rv = self.invoke(ctx)
File "C:\Users\28636\AppData\Local\Programs\Python\Python39\lib\site-packages\click\core.py", line 1434, in invoke
return ctx.invoke(self.callback, **ctx.params)
File "C:\Users\28636\AppData\Local\Programs\Python\Python39\lib\site-packages\click\core.py", line 783, in invoke
return __callback(*args, **kwargs)
File "E:\GAN\stylegan3-main\train.py", line 281, in main
launch_training(c=c, desc=desc, outdir=opts.outdir, dry_run=opts.dry_run)
File "E:\GAN\stylegan3-main\train.py", line 96, in launch_training
subprocess_fn(rank=0, c=c, temp_dir=temp_dir)
File "E:\GAN\stylegan3-main\train.py", line 47, in subprocess_fn
training_loop.training_loop(rank=rank, **c)
File "E:\GAN\stylegan3-main\training\training_loop.py", line 168, in training_loop
img = misc.print_module_summary(G, [z, c])
File "E:\GAN\stylegan3-main\torch_utils\misc.py", line 216, in print_module_summary
outputs = module(*inputs)
File "C:\Users\28636\AppData\Local\Programs\Python\Python39\lib\site-packages\torch\nn\modules\module.py", line 1538, in _call_impl
result = forward_call(*args, **kwargs)
File "E:\GAN\stylegan3-main\training\networks_stylegan3.py", line 511, in forward
ws = self.mapping(z, c, truncation_psi=truncation_psi, truncation_cutoff=truncation_cutoff, update_emas=update_emas) File "C:\Users\28636\AppData\Local\Programs\Python\Python39\lib\site-packages\torch\nn\modules\module.py", line 1538, in _call_impl
result = forward_call(*args, **kwargs)
File "E:\GAN\stylegan3-main\training\networks_stylegan3.py", line 151, in forward
x = getattr(self, f'fc{idx}')(x)
File "C:\Users\28636\AppData\Local\Programs\Python\Python39\lib\site-packages\torch\nn\modules\module.py", line 1538, in _call_impl
result = forward_call(*args, **kwargs)
File "E:\GAN\stylegan3-main\training\networks_stylegan3.py", line 100, in forward
x = bias_act.bias_act(x, b, act=self.activation)
File "E:\GAN\stylegan3-main\torch_utils\ops\bias_act.py", line 84, in bias_act
if impl == 'cuda' and x.device.type == 'cuda' and _init():
File "E:\GAN\stylegan3-main\torch_utils\ops\bias_act.py", line 41, in _init
_plugin = custom_ops.get_plugin(
File "E:\GAN\stylegan3-main\torch_utils\custom_ops.py", line 136, in get_plugin
torch.utils.cpp_extension.load(name=module_name, build_directory=cached_build_dir,
File "C:\Users\28636\AppData\Local\Programs\Python\Python39\lib\site-packages\torch\utils\cpp_extension.py", line 1284, in load
return _jit_compile(
File "C:\Users\28636\AppData\Local\Programs\Python\Python39\lib\site-packages\torch\utils\cpp_extension.py", line 1535, in _jit_compile
return _import_module_from_library(name, build_directory, is_python_module)
File "C:\Users\28636\AppData\Local\Programs\Python\Python39\lib\site-packages\torch\utils\cpp_extension.py", line 1929, in _import_module_from_library
module = importlib.util.module_from_spec(spec)
File "<frozen importlib._bootstrap>", line 565, in module_from_spec
File "<frozen importlib._bootstrap_external>", line 1108, in create_module
File "<frozen importlib._bootstrap>", line 228, in _call_with_frames_removed
ImportError: DLL load failed while importing bias_act_plugin: 找不到指定的模块。
StyleGAN 用风格(style)来影响人脸的姿态、身份特征等,用噪声 ( noise ) 来影响头发丝、皱纹、肤色等细节部分。
StyleGAN 的网络结构包含两个部分,第一个是Mapping network,即下图 (b)中的左部分,由隐藏变量 z 生成 中间隐藏变量 w的过程,这个 w 就是用来控制生成图像的style,即风格,为什么要多此一举将 z 变成 w 呢,后面会详细讲到。 第二个是Synthesis network,它的作用是生成图像,创新之处在于给每一层子网络都喂了 A 和 B,A 是由 w 转换得到的仿射变换,用于控制生成图像的风格(1.2),B 是转换后的随机噪声,用于丰富生成图像的细节,例如:雀斑、皱纹(1.4),即每个卷积层都能根据输入的A来调整"style",通过B来调整细节。整个网络结构还是保持了 PG-GAN (progressive growing GAN) 的结构。