Update generator_firstorder.py

fixed bug of importing ppgan model
own
kongdebug 3 years ago committed by GitHub
parent 36ff6dd67d
commit 9deb8cf0a7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 4
      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()

Loading…
Cancel
Save