def __init__()

in empose/nn/models.py [0:0]

• 17 lines of code
• 7 McCabe index (conditional complexity)

    def __init__(self, config, smpl_model=None):
        super(BaseModel, self).__init__()
        self.n_markers = config.n_markers if hasattr(config, 'n_markers') and config.n_markers > -1 else C.N_TRACKERS_WO_ROOT
        self.config = config
        self.n_frames = config.window_size
        self.smpl = smpl_model

        self.estimate_shape = self.config.m_estimate_shape
        self.shape_avg = config.m_average_shape

        self.fk_loss_weight = config.m_fk_loss
        self.do_fk = self.fk_loss_weight > 0.0
        if self.do_fk:
            assert self.smpl is not None
            assert self.estimate_shape or isinstance(self, IterativeErrorFeedback)

        self.shape_weight = config.m_shape_loss_weight if hasattr(config, 'm_shape_loss_weight') else 1.0
        self.pose_weight = config.m_pose_loss_weight if hasattr(config, 'm_pose_loss_weight') else 1.0

        self.set_input_output_size()
        self.create_model()