Browse Source

Update generator_starganv2.py

fixed bug of importing ppgan model
kongdebug 3 years ago
parent
commit
d4aa2b9b53
1 changed files with 2 additions and 3 deletions
  1. 2 3
      paddlers/models/ppgan/models/generators/generator_starganv2.py

+ 2 - 3
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"