Yuliang commited on
Commit
c5b04dc
1 Parent(s): eea1e27

Update apps/infer.py

Browse files
Files changed (1) hide show
  1. apps/infer.py +1 -0
apps/infer.py CHANGED
@@ -29,6 +29,7 @@ from lib.dataset.mesh_util import (
29
  )
30
 
31
  from lib.dataset.TestDataset import TestDataset
 
32
  from lib.net.local_affine import LocalAffine
33
  from pytorch3d.structures import Meshes
34
  from apps.ICON import ICON
 
29
  )
30
 
31
  from lib.dataset.TestDataset import TestDataset
32
+ from lib.common.render import query_color
33
  from lib.net.local_affine import LocalAffine
34
  from pytorch3d.structures import Meshes
35
  from apps.ICON import ICON