build_graph/data/epic.py [168:187]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def __init__(self, root):
        super().__init__(root)

        vid_lengths = self.annotations['vid_lengths']
        frames = []
        for v_id in vid_lengths:
            frames += [(v_id, f_id) for f_id in range(1, vid_lengths[v_id]+1)]
        self.frames = frames
        self.transform = util.default_transform('val')

        self.keys = ['%s/%d'%(v_id, f_id) for v_id, f_id in self.frames]
        self.keys = np.array(self.keys, dtype='S')

    def __getitem__(self, index):
        img = self.load_image(self.frames[index])
        img = self.transform(img)
        return {'frame': img}

    def __len__(self):
        return len(self.frames)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



build_graph/data/gtea.py [177:196]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def __init__(self, root):
        super().__init__(root)

        vid_lengths = self.annotations['vid_lengths']
        frames = []
        for v_id in vid_lengths:
            frames += [(v_id, f_id) for f_id in range(1, vid_lengths[v_id]+1)]
        self.frames = frames
        self.transform = util.default_transform('val')

        self.keys = ['%s/%d'%(v_id, f_id) for v_id, f_id in self.frames]
        self.keys = np.array(self.keys, dtype='S')

    def __getitem__(self, index):
        img = self.load_image(self.frames[index])
        img = self.transform(img)
        return {'frame': img}

    def __len__(self):
        return len(self.frames)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



