Skip to content

Commit

Permalink
[Fix] Fix several typos in apps (#120)
Browse files Browse the repository at this point in the history
* fix applications

* fix lint

Co-authored-by: yangyifei <PJLAB\yangyifei@shai14001042l.pjlab.org>
  • Loading branch information
plyfager and yangyifei committed Sep 16, 2021
1 parent a9c0ac4 commit a9a427b
Show file tree
Hide file tree
Showing 4 changed files with 25 additions and 14 deletions.
14 changes: 8 additions & 6 deletions apps/conditional_interpolate.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
import argparse
import os
import sys

import mmcv
import torch
Expand All @@ -8,15 +9,16 @@
from mmcv.runner import load_checkpoint
from torchvision.utils import save_image

from mmgen.apis import set_random_seed
from mmgen.core.evaluation import slerp
from mmgen.models import build_model
from mmgen.models.architectures import BigGANDeepGenerator, BigGANGenerator
from mmgen.models.architectures.common import get_module_device

# yapf: disable
sys.path.append(os.path.abspath(os.path.join(__file__, '../..'))) # isort:skip # noqa

from mmgen.apis import set_random_seed # isort:skip # noqa
from mmgen.core.evaluation import slerp # isort:skip # noqa
from mmgen.models import build_model # isort:skip # noqa
from mmgen.models.architectures import BigGANDeepGenerator, BigGANGenerator # isort:skip # noqa
from mmgen.models.architectures.common import get_module_device # isort:skip # noqa

# yapf: enable

_default_embedding_name = dict(
BigGANGenerator='shared_embedding',
Expand Down
11 changes: 7 additions & 4 deletions apps/interpolate_sample.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
import argparse
import os
import sys

import mmcv
import numpy as np
Expand All @@ -8,13 +9,15 @@
from mmcv.runner import load_checkpoint
from torchvision.utils import save_image

from mmgen.apis import set_random_seed
from mmgen.core.evaluation import slerp
from mmgen.models import build_model

# yapf: disable
sys.path.append(os.path.abspath(os.path.join(__file__, '../..'))) # isort:skip # noqa

from mmgen.apis import set_random_seed # isort:skip # noqa
from mmgen.core.evaluation import slerp # isort:skip # noqa
from mmgen.models import build_model # isort:skip # noqa

# yapf: enable


def parse_args():
parser = argparse.ArgumentParser(
Expand Down
2 changes: 1 addition & 1 deletion apps/modified_sefa.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ def calc_eigens(args, state_dict):
'--degree',
type=float,
nargs='+',
default=2,
default=[2.],
help='scalar factors for moving latent vectors along eigenvector',
)
parser.add_argument(
Expand Down
12 changes: 9 additions & 3 deletions apps/stylegan_projector.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
"""
import argparse
import os
import sys

import mmcv
import numpy as np
Expand All @@ -19,9 +20,14 @@
from torchvision.utils import save_image
from tqdm import tqdm

from mmgen.apis import set_random_seed
from mmgen.models import build_model
from mmgen.models.architectures.lpips import PerceptualLoss
# yapf: disable
sys.path.append(os.path.abspath(os.path.join(__file__, '../..'))) # isort:skip # noqa

from mmgen.apis import set_random_seed # isort:skip # noqa
from mmgen.models import build_model # isort:skip # noqa
from mmgen.models.architectures.lpips import PerceptualLoss # isort:skip # noqa

# yapf: enable


def parse_args():
Expand Down

0 comments on commit a9a427b

Please sign in to comment.