Arnaudding001 commited on
Commit
d413e42
·
1 Parent(s): 97f1ec6

Update util.py

Browse files
Files changed (1) hide show
  1. util.py +3 -3
util.py CHANGED
@@ -11,9 +11,9 @@ from torch.nn import functional as F
11
  from torch import autograd
12
  from torch.nn import init
13
  import torchvision.transforms as transforms
14
- from model.stylegan.op import conv2d_gradfix
15
- from model.encoder.encoders.psp_encoders import GradualStyleEncoder
16
- from model.encoder.align_all_parallel import get_landmark
17
 
18
  def visualize(img_arr, dpi):
19
  plt.figure(figsize=(10,10),dpi=dpi)
 
11
  from torch import autograd
12
  from torch.nn import init
13
  import torchvision.transforms as transforms
14
+ import stylegan_op_conv2d_gradfix
15
+ from encoder_encoders_psp_encoders import GradualStyleEncoder
16
+ from encoder_align_all_parallel import get_landmark
17
 
18
  def visualize(img_arr, dpi):
19
  plt.figure(figsize=(10,10),dpi=dpi)