nets/__init__.py

from .smplx_face import TrainWrapper as s2g_face
from .smplx_body_vq import TrainWrapper as s2g_body_vq
from .smplx_body_pixel import s2g_body_pixel
from .body_ae import TrainWrapper as s2g_body_ae
from .LS3DCG import TrainWrapper as LS3DCG
from .base import TrainWrapperBaseClass

from .utils import normalize, denormalize