update ehf 230804

open-mmlab · Aug 4, 2023 · 32c0802 · 32c0802
1 parent b8f30f1
commit 32c0802
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/mmhuman3d/data/data_converters/__init__.py b/mmhuman3d/data/data_converters/__init__.py
@@ -12,7 +12,7 @@
 from .dynacam import DynacamConverter
 from .eft import EftConverter
 from .egobody import EgobodyConverter
-from .ehf import EHFConverter
+from .ehf import EhfConverter
 from .expose_curated_fits import ExposeCuratedFitsConverter
 from .expose_spin_smplx import ExposeSPINSMPLXConverter
 from .ffhq_flame import FFHQFlameConverter
@@ -88,7 +88,7 @@
     'ExposeSPINSMPLXConverter',
     'FreihandConverter',
     'StirlingConverter',
-    'EHFConverter',
+    'EhfConverter',
     'CliffConverter',
     'SynbodyConverter',
     'RenbodyConverter',

diff --git a/mmhuman3d/data/data_converters/ehf.py b/mmhuman3d/data/data_converters/ehf.py
@@ -12,7 +12,7 @@
 
 
 @DATA_CONVERTERS.register_module()
-class EHFConverter(BaseModeConverter):
+class EhfConverter(BaseModeConverter):
     """EHF dataset 'Expressive Hands and Face' More details can be found on the
     website:
 

diff --git a/tools/convert_datasets.py b/tools/convert_datasets.py
@@ -102,6 +102,10 @@
         type='Cimi4dConverter',  # real
         prefix='cimi4d',
         modes=['train']),
+    ehf=dict(
+        type='EhfConverter',  # real'
+        prefix='ehf',
+        modes=['val']),
     hsc4d=dict(
         type='Hsc4dConverter',  # real, in progress
         prefix='hsc4d',