Skip to content

Commit

Permalink
add stylegan2_bilinear_arch
Browse files Browse the repository at this point in the history
  • Loading branch information
xinntao committed Feb 14, 2022
1 parent 77df6e4 commit 47983e1
Show file tree
Hide file tree
Showing 2 changed files with 615 additions and 2 deletions.
4 changes: 2 additions & 2 deletions gfpgan/archs/gfpgan_bilinear_arch.py
Original file line number Diff line number Diff line change
@@ -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):
Expand Down
Loading

0 comments on commit 47983e1

Please sign in to comment.