From d4aa2b9b536b8d1358eed1622c45c82084feebc2 Mon Sep 17 00:00:00 2001 From: kongdebug <52785738+kongdebug@users.noreply.github.com> Date: Thu, 24 Mar 2022 20:25:34 +0800 Subject: [PATCH] Update generator_starganv2.py fixed bug of importing ppgan model --- .../models/ppgan/models/generators/generator_starganv2.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/paddlers/models/ppgan/models/generators/generator_starganv2.py b/paddlers/models/ppgan/models/generators/generator_starganv2.py index a8cef90..7d39de5 100644 --- a/paddlers/models/ppgan/models/generators/generator_starganv2.py +++ b/paddlers/models/ppgan/models/generators/generator_starganv2.py @@ -9,9 +9,8 @@ from .builder import GENERATORS import numpy as np import math -from ppgan.modules.wing import CoordConvTh, ConvBlock, HourGlass, preprocess - -from ppgan.utils.download import get_path_from_url +from ...modules.wing import CoordConvTh, ConvBlock, HourGlass, preprocess +from ...utils.download import get_path_from_url FAN_WEIGHT_URL = "https://paddlegan.bj.bcebos.com/models/wing.pdparams"