Spaces:
Runtime error
Runtime error
Update apps/infer.py
Browse files- 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
|