-
Notifications
You must be signed in to change notification settings - Fork 0
/
template_code.py
35 lines (30 loc) · 1.23 KB
/
template_code.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
from read_datasetBreakfast import load_data, read_mapping_dict
import os
COMP_PATH = ''
'''
training to load train set
test to load test set
'''
split = 'test'
#split = 'test'
train_split = os.path.join(COMP_PATH, 'splits/train.split1.bundle') #Train Split
test_split = os.path.join(COMP_PATH, 'splits/test.split1.bundle') #Test Split
GT_folder = os.path.join(COMP_PATH, 'groundTruth/') #Ground Truth Labels for each training video
DATA_folder = os.path.join(COMP_PATH, 'data/') #Frame I3D features for all videos
mapping_loc = os.path.join(COMP_PATH, 'splits/mapping_bf.txt')
actions_dict = read_mapping_dict(mapping_loc)
if split == 'training':
data_feat, data_labels = load_data( train_split, actions_dict, GT_folder, DATA_folder, datatype = split) #Get features and labels
if split == 'test':
data_feat = load_data( test_split, actions_dict, GT_folder, DATA_folder, datatype = split) #Get features only
'''
Write Code Below
Pointers
Need to load the segments.txt file for segments for test videos
Output the CSV in correct format as shown in Evaluation Section
Id corresponds to the segments in order.
Example - 30-150 = Id 0
150-428 = Id 1
428-575 = Id 2
Category is the Class of the Predicted Action
'''