guardiancc commited on
Commit
0e0a3f2
·
verified ·
1 Parent(s): 33e6f62

Update engine.py

Browse files
Files changed (1) hide show
  1. engine.py +2 -0
engine.py CHANGED
@@ -20,6 +20,8 @@ from tqdm import tqdm as loader
20
  import cv2
21
  import torch
22
  from basicsr.archs.srvgg_arch import SRVGGNetCompact
 
 
23
 
24
  from liveportrait.config.argument_config import ArgumentConfig
25
  from liveportrait.utils.camera import get_rotation_matrix
 
20
  import cv2
21
  import torch
22
  from basicsr.archs.srvgg_arch import SRVGGNetCompact
23
+ from gfpgan.utils import GFPGANer
24
+ from realesrgan.utils import RealESRGANer
25
 
26
  from liveportrait.config.argument_config import ArgumentConfig
27
  from liveportrait.utils.camera import get_rotation_matrix