def from_sample_list()

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

• 27 lines of code
• 2 McCabe index (conditional complexity)

    def from_sample_list(samples):
        """Collect a set of `RealSample`s into a batch."""
        ids = []
        seq_lengths = []
        marker_pos = []
        marker_ori = []
        marker_masks = []
        poses = []
        shapes = []
        trans = []
        offset_t = []
        offset_r = []
        for sample in samples:
            ids.append(sample.id)
            seq_lengths.append(sample.smpl_poses.shape[0])
            marker_pos.append(sample.marker_pos_real)
            marker_ori.append(sample.marker_ori_real)
            marker_masks.append(sample.marker_masks)
            poses.append(sample.smpl_poses)
            shapes.append(sample.smpl_shape)
            trans.append(sample.smpl_trans)
            offset_t.append(sample.offset_means)
            offset_r.append(sample.offset_r)
        return RealBatch(ids, torch.from_numpy(np.array(seq_lengths)), pad_sequence(poses, batch_first=True),
                         pad_sequence(shapes, batch_first=True), pad_sequence(trans, batch_first=True),
                         pad_sequence(marker_pos, batch_first=True), pad_sequence(marker_ori, batch_first=True),
                         pad_sequence(marker_masks, batch_first=True),
                         torch.stack(offset_t), torch.stack(offset_r))