瀏覽代碼

Update generator_firstorder.py

fixed bug of importing ppgan model
kongdebug 3 年之前
父節點
當前提交
9deb8cf0a7
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      paddlers/models/ppgan/models/generators/generator_firstorder.py

+ 2 - 2
paddlers/models/ppgan/models/generators/generator_firstorder.py

@@ -7,13 +7,13 @@ import paddle
 import paddle.nn.functional as F
 from paddle import nn
 
-from ppgan.models.generators.builder import GENERATORS
+from ...models.generators.builder import GENERATORS
 from .occlusion_aware import OcclusionAwareGenerator
 from ...modules.first_order import make_coordinate_grid, ImagePyramide, detach_kp
 from ...modules.keypoint_detector import KPDetector
 
 import paddle.vision.models.vgg as vgg
-from ppgan.utils.download import get_path_from_url
+from ...utils.download import get_path_from_url
 
 
 @GENERATORS.register()