Method load_regressor

detrsmpl/models/heads/expose_head.py:201–210

Build regressor for ExPose Head.

(self,
                       input_feat_dim: int = 2048,
                       param_mean: torch.Tensor = None,
                       regressor_cfg: dict = None)

        super().__init__(init_cfg)

    def load_regressor(self,
                       input_feat_dim: int = 2048,
                       param_mean: torch.Tensor = None,
                       regressor_cfg: dict = None):
        """Build regressor for ExPose Head."""
        param_dim = param_mean.numel()
        regressor = MLP(input_feat_dim + param_dim, param_dim, **regressor_cfg)
        self.regressor = IterativeRegression(regressor,
                                             param_mean,
                                             num_stages=3)

    def load_param_decoder(self, mean_poses_dict):
        """Build decoders for each pose."""