def __init__()

in dataset/dataset_downstream.py [0:0]

• 23 lines of code
• 5 McCabe index (conditional complexity)

    def __init__(self, root, split='split_1', mode='train', data_source='ucf',**kwargs):
        super(DownstreamClipFolderDataset, self).__init__()

        self.dataset_root = root
        # data frame root
        self.dataset_frame_root = os.path.join(self.dataset_root, 'rawframes')
        assert os.path.exists(self.dataset_frame_root)

        # data list file
        assert mode in ('train', 'val', 'test')
        assert split in ('split_1', 'split_2', 'split_3')
        self.data_source = data_source

        if data_source == 'ucf':
            self.dataset_list_file = os.path.join(self.dataset_root, 'ucfTrainTestlist',
                                                  'ucf101_' + mode + '_' + split + '_rawframes' + '.txt')
        elif data_source == 'hmdb':
            self.dataset_list_file = os.path.join(self.dataset_root, 'testTrainMulti_7030_splits',
                                                  'hmdb51_' + mode + '_' + split + '_rawframes' + '.txt')
        elif data_source == 'sthv2':
            self.dataset_list_file = os.path.join(self.dataset_root,
                                                  'sthv2_' + mode + '_list' + '_rawframes' + '.txt')
        elif data_source == 'anet':
            self.dataset_list_file = os.path.join(self.dataset_root,
                                                  'anet_' + mode + '_video' + '.txt')
        assert os.path.exists(self.dataset_list_file)

        # load vid samples
        self.samples = self._load_list(self.dataset_list_file)
        self.transform = None