def __init__(self, app):
    self.app = app
    self.masking = True
    if (app == 'block_construction'):
        self.file_directory = os.path.join(DATASET_DIR, 'BlockConstruction')
        self.trace_file = os.path.join(self.file_directory, 'traces6.npy')
        self.obs_label_list = {'n': 0, 'r': 1, 'rr': 2, 'rrr': 3, 'g': 4, 'gb': 5, 'bg': 6, 'b': 7}
        self.act_label_list = {'N': 0, 'R': 1, 'G': 2, 'B': 3}
        self.tsm = {'filename': 'c_backbone_tsm_1_bn16', 'bottleneck': 16}
        self.wrn = {'filename': 'c_backbone_wrn_2_bn16', 'bottleneck': 16}
        self.i3d = {'filename': 'c_backbone_i3d_1_bn8', 'bottleneck': 8}
        self.vgg = {'filename': 'c_backbone_vgg_2_bn32', 'bottleneck': 32}
    elif (app == 'block_construction_timed'):
        self.file_directory = os.path.join(DATASET_DIR, 'BlockConstructionTimed')
        self.trace_file = os.path.join(self.file_directory, 'traces6.npy')
        self.obs_label_list = {'n': 0, 'r': 1, 'rr': 2, 'rrr': 3, 'g': 4, 'gb': 5, 'bg': 6, 'b': 7}
        self.act_label_list = {'N': 0, 'R': 1, 'G': 2, 'B': 3}
        self.tsm = {'filename': 'c_backbone_tsm_1_bn16', 'bottleneck': 16}
        self.wrn = {'filename': 'c_backbone_wrn_0_bn16', 'bottleneck': 16}
        self.i3d = {'filename': 'c_backbone_i3d_1_bn16', 'bottleneck': 16}
        self.vgg = {'filename': 'c_backbone_vgg_0_bn32', 'bottleneck': 32}
    elif (app == 'ikea'):
        self.file_directory = os.path.join(DATASET_DIR, 'IKEA_fa')
        label_path = os.path.join(*[self.file_directory, 'frames', 'train'])
        self.obs_label_list = {k: v for (v, k) in enumerate(os.listdir(label_path))}
        self.act_label_list = None
        self.masking = False
        self.tsm = {'filename': 'c_backbone_tsm_0', 'bottleneck': 64}
        self.wrn = {'filename': 'c_backbone_wrn_0', 'bottleneck': 64}
        self.i3d = {'filename': 'c_backbone_i3d_0', 'bottleneck': 64}
        self.vgg = {'filename': 'c_backbone_vgg_0', 'bottleneck': 64}
    elif (app == 'crepe_action'):
        self.file_directory = os.path.join(DATASET_DIR, 'CrepeAction')
        label_path = os.path.join(*[self.file_directory, 'frames', 'train'])
        self.obs_label_list = {k: v for (v, k) in enumerate(sorted(os.listdir(label_path)))}
        self.act_label_list = None
        self.masking = True
        self.tsm = {'filename': 'c_backbone_tsm_1', 'bottleneck': 64}
        self.wrn = {'filename': 'c_backbone_wrn_2', 'bottleneck': 64}
        self.i3d = {'filename': 'c_backbone_i3d_1', 'bottleneck': 64}
        self.vgg = {'filename': 'c_backbone_vgg_0', 'bottleneck': 64}
    elif (app == 'crepe_recipe'):
        self.file_directory = os.path.join(DATASET_DIR, 'CrepeRecipe')
        label_path = os.path.join(*[self.file_directory, 'frames', 'train'])
        self.obs_label_list = {k: v for (v, k) in enumerate(sorted(os.listdir(label_path)))}
        self.act_label_list = None
        self.masking = True
        self.tsm = {'filename': 'c_backbone_tsm_1', 'bottleneck': 64}
        self.wrn = {'filename': 'c_backbone_wrn_2', 'bottleneck': 64}
        self.i3d = {'filename': 'c_backbone_i3d_1', 'bottleneck': 64}
        self.vgg = {'filename': 'c_backbone_vgg_0', 'bottleneck': 64}
    self.num_labels = len(self.obs_label_list)
def __init__(self, app):
    self.app = app
    self.masking = True
    if (app == 'block_construction'):
        self.file_directory = os.path.join(DATASET_DIR, 'BlockConstruction')
        self.trace_file = os.path.join(self.file_directory, 'traces6.npy')
        self.obs_label_list = {'n': 0, 'r': 1, 'rr': 2, 'rrr': 3, 'g': 4, 'gb': 5, 'bg': 6, 'b': 7}
        self.act_label_list = {'N': 0, 'R': 1, 'G': 2, 'B': 3}
        self.tsm = {'filename': 'c_backbone_tsm_1_bn16', 'bottleneck': 16}
        self.wrn = {'filename': 'c_backbone_wrn_2_bn16', 'bottleneck': 16}
        self.i3d = {'filename': 'c_backbone_i3d_1_bn8', 'bottleneck': 8}
        self.vgg = {'filename': 'c_backbone_vgg_2_bn32', 'bottleneck': 32}
    elif (app == 'block_construction_timed'):
        self.file_directory = os.path.join(DATASET_DIR, 'BlockConstructionTimed')
        self.trace_file = os.path.join(self.file_directory, 'traces6.npy')
        self.obs_label_list = {'n': 0, 'r': 1, 'rr': 2, 'rrr': 3, 'g': 4, 'gb': 5, 'bg': 6, 'b': 7}
        self.act_label_list = {'N': 0, 'R': 1, 'G': 2, 'B': 3}
        self.tsm = {'filename': 'c_backbone_tsm_1_bn16', 'bottleneck': 16}
        self.wrn = {'filename': 'c_backbone_wrn_0_bn16', 'bottleneck': 16}
        self.i3d = {'filename': 'c_backbone_i3d_1_bn16', 'bottleneck': 16}
        self.vgg = {'filename': 'c_backbone_vgg_0_bn32', 'bottleneck': 32}
    elif (app == 'ikea'):
        self.file_directory = os.path.join(DATASET_DIR, 'IKEA_fa')
        label_path = os.path.join(*[self.file_directory, 'frames', 'train'])
        self.obs_label_list = {k: v for (v, k) in enumerate(os.listdir(label_path))}
        self.act_label_list = None
        self.masking = False
        self.tsm = {'filename': 'c_backbone_tsm_0', 'bottleneck': 64}
        self.wrn = {'filename': 'c_backbone_wrn_0', 'bottleneck': 64}
        self.i3d = {'filename': 'c_backbone_i3d_0', 'bottleneck': 64}
        self.vgg = {'filename': 'c_backbone_vgg_0', 'bottleneck': 64}
    elif (app == 'crepe_action'):
        self.file_directory = os.path.join(DATASET_DIR, 'CrepeAction')
        label_path = os.path.join(*[self.file_directory, 'frames', 'train'])
        self.obs_label_list = {k: v for (v, k) in enumerate(sorted(os.listdir(label_path)))}
        self.act_label_list = None
        self.masking = True
        self.tsm = {'filename': 'c_backbone_tsm_1', 'bottleneck': 64}
        self.wrn = {'filename': 'c_backbone_wrn_2', 'bottleneck': 64}
        self.i3d = {'filename': 'c_backbone_i3d_1', 'bottleneck': 64}
        self.vgg = {'filename': 'c_backbone_vgg_0', 'bottleneck': 64}
    elif (app == 'crepe_recipe'):
        self.file_directory = os.path.join(DATASET_DIR, 'CrepeRecipe')
        label_path = os.path.join(*[self.file_directory, 'frames', 'train'])
        self.obs_label_list = {k: v for (v, k) in enumerate(sorted(os.listdir(label_path)))}
        self.act_label_list = None
        self.masking = True
        self.tsm = {'filename': 'c_backbone_tsm_1', 'bottleneck': 64}
        self.wrn = {'filename': 'c_backbone_wrn_2', 'bottleneck': 64}
        self.i3d = {'filename': 'c_backbone_i3d_1', 'bottleneck': 64}
        self.vgg = {'filename': 'c_backbone_vgg_0', 'bottleneck': 64}
    self.num_labels = len(self.obs_label_list)