FLAME as a layer model constructor.
(self, *args, **kwargs)
| 2045 | |
| 2046 | class FLAMELayer(FLAME): |
| 2047 | def __init__(self, *args, **kwargs) -> None: |
| 2048 | """FLAME as a layer model constructor.""" |
| 2049 | super(FLAMELayer, self).__init__(create_betas=False, |
| 2050 | create_expression=False, |
| 2051 | create_global_orient=False, |
| 2052 | create_neck_pose=False, |
| 2053 | create_jaw_pose=False, |
| 2054 | create_leye_pose=False, |
| 2055 | create_reye_pose=False, |
| 2056 | *args, |
| 2057 | **kwargs) |
| 2058 | |
| 2059 | def forward(self, |
| 2060 | betas: Optional[Tensor] = None, |