diff --git a/face_swapper/src/generator.py b/face_swapper/src/generator.py index e93a668..2f92bc3 100644 --- a/face_swapper/src/generator.py +++ b/face_swapper/src/generator.py @@ -1,7 +1,9 @@ +from typing import Tuple + import torch import torch.nn as nn -from .typing import IDEmbedding, TargetAttributes, Tensor, Tuple +from .typing import IDEmbedding, TargetAttributes, Tensor class AdaptiveEmbeddingIntegrationNetwork(nn.Module): diff --git a/face_swapper/src/helper.py b/face_swapper/src/helper.py index 6f42cd2..954e0b6 100644 --- a/face_swapper/src/helper.py +++ b/face_swapper/src/helper.py @@ -1,8 +1,9 @@ import configparser +from typing import Tuple import torch -from .typing import Tensor, Tuple +from .typing import Tensor CONFIG = configparser.ConfigParser() CONFIG.read('config.ini') diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py index 12d740b..6070d3c 100644 --- a/face_swapper/src/training.py +++ b/face_swapper/src/training.py @@ -1,5 +1,6 @@ import configparser import os +from typing import Tuple import cv2 import numpy @@ -17,7 +18,7 @@ from .data_loader import DataLoaderVGG, read_image from .discriminator import MultiscaleDiscriminator from .generator import AdaptiveEmbeddingIntegrationNetwork from .helper import L2_loss, hinge_loss -from .typing import Batch, DiscriminatorOutputs, IDEmbedding, LossDict, TargetAttributes, Tensor, Tuple +from .typing import Batch, DiscriminatorOutputs, IDEmbedding, LossDict, TargetAttributes, Tensor CONFIG = configparser.ConfigParser() CONFIG.read('config.ini')