Update generator_starganv2.py

fixed bug of importing ppgan model
3 years ago · d4aa2b9b53
parent 9deb8cf0a7
commit d4aa2b9b53
1 changed files with 2 additions and 3 deletions
--- 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"