def get_inputs()

in empose/data/data.py [0:0]

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

    def get_inputs(self, sf=None, ef=None, **kwargs):
        """Returns the inputs that we feed to a model."""
        inputs_ = {'marker_pos': None, 'marker_oris': None, 'marker_normals': None, 'joints': None}

        if self.marker_pos_noisy is not None:
            inputs_['marker_pos'] = self.marker_pos_noisy.detach()[:, sf:ef]
        elif self.marker_pos_synth is not None:
            inputs_['marker_pos'] = self.marker_pos_synth.detach()[:, sf:ef]

        if self.marker_ori_noisy is not None:
            inputs_['marker_oris'] = self.marker_ori_noisy.detach()[:, sf:ef]
        elif self.marker_ori_synth is not None:
            inputs_['marker_oris'] = self.marker_ori_synth.detach()[:, sf:ef]

        if self.marker_normal_noisy is not None:
            inputs_['marker_normals'] = self.marker_normal_noisy.detach()[:, sf:ef]
        elif self.marker_normal_synth is not None:
            inputs_['marker_normals'] = self.marker_normal_synth.detach()[:, sf:ef]

        joints = self.joints_gt

        inputs_['joints'] = joints[:, sf:ef]
        inputs_['offset_t'] = self.offset_t_augmented
        inputs_['offset_r'] = self.offset_r_augmented
        inputs_['marker_masks'] = None

        return inputs_