build_graph/data/epic.py [107:128]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def __init__(self, root, split, clip_len):
        super().__init__(root)

        self.split = split
        self.clip_len = clip_len

        self.train_data = self.parse_data_for_split(self.train_vids)
        self.val_data = self.parse_data_for_split(self.val_vids)
        self.data = self.train_data if self.split=='train' else self.val_data
        print (f'Train data: {len(self.train_data)} | Val data: {len(self.val_data)}')
        
        # self.clip_transform = util.clip_transform(self.split, self.clip_len)    

    def parse_data_for_split(self, videos):
        videos = set(videos)
        clips = []
        for entry in self.interactions:

            if entry['v_id'] not in videos:
                continue

            clip = dict(entry)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



build_graph/data/gtea.py [116:137]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def __init__(self, root, split, clip_len):
        super().__init__(root)

        self.split = split
        self.clip_len = clip_len

        self.train_data = self.parse_data_for_split(self.train_vids)
        self.val_data = self.parse_data_for_split(self.val_vids)
        self.data = self.train_data if self.split=='train' else self.val_data
        print (f'Train data: {len(self.train_data)} | Val data: {len(self.val_data)}')
        
        # self.clip_transform = util.clip_transform(self.split, self.clip_len)    

    def parse_data_for_split(self, videos):
        videos = set(videos)
        clips = []
        for entry in self.interactions:

            if entry['v_id'] not in videos:
                continue

            clip = dict(entry)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -