Method transform_semantic

data/inference_dataset.py:148–153

(self, semantic, frame_index)

        }

    def transform_semantic(self, semantic, frame_index):
        # 500, 73
        index = self.obtain_seq_index(frame_index, semantic.shape[0])
        coeff_3dmm = semantic[index, ...]
        # 27, 73
        return torch.Tensor(coeff_3dmm).permute(1, 0)

    def obtain_seq_index(self, index, num_frames):
        seq = list(range(index - self.semantic_radius, index + self.semantic_radius + 1))