add stylegan2_bilinear_arch

TencentARC · Feb 14, 2022 · 47983e1 · 47983e1
1 parent 77df6e4
commit 47983e1
Show file tree

Hide file tree

Showing 2 changed files with 615 additions and 2 deletions.
diff --git a/gfpgan/archs/gfpgan_bilinear_arch.py b/gfpgan/archs/gfpgan_bilinear_arch.py
@@ -1,12 +1,12 @@
 import math
 import random
 import torch
-from basicsr.archs.stylegan2_bilinear_arch import (ConvLayer, EqualConv2d, EqualLinear, ResBlock, ScaledLeakyReLU,
- StyleGAN2GeneratorBilinear)
 from basicsr.utils.registry import ARCH_REGISTRY
 from torch import nn
 
 from .gfpganv1_arch import ResUpBlock
+from .stylegan2_bilinear_arch import (ConvLayer, EqualConv2d, EqualLinear, ResBlock, ScaledLeakyReLU,
+ StyleGAN2GeneratorBilinear)
 
 
 class StyleGAN2GeneratorBilinearSFT(StyleGAN2GeneratorBilinear):