re-format SMC reader import to reduce error

open-mmlab · May 30, 2022 · 26aeaad · 26aeaad
1 parent d1d1fce
commit 26aeaad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmhuman3d/data/datasets/pipelines/loading.py b/mmhuman3d/data/datasets/pipelines/loading.py
@@ -4,7 +4,7 @@
 import mmcv
 import numpy as np
 
-import mmhuman3d.data.data_structures as data_structures
+from mmhuman3d.data.data_structures.smc_reader import SMCReader
 from ..builder import PIPELINES
 
 
@@ -50,7 +50,7 @@ def __call__(self, results):
             assert 'image_id' in results, 'Load image from .smc, ' \
                                           'but image_id is not provided.'
             device, device_id, frame_id = results['image_id']
-            smc_reader = data_structures.SMCReader(filename)
+            smc_reader = SMCReader(filename)
             img = smc_reader.get_color(
                 device, device_id, frame_id, disable_tqdm=True)
             img = img.squeeze()  # (1, H, W, 3) -> (H, W, 3)