Exemplos de ANetDB em Python

Exemplo n.º 1

def parse_activitynet_splits(version):
    db = ANetDB.get_db(version)
    train_instance = db.get_subset_instance('training')
    val_instance = db.get_subset_instance('validation')
    test_instance = db.get_subset_videos('testing')

    splits = []

    train_list = [(x.name, x.num_label) for x in train_instance]
    val_list = [(x.name, x.num_label) for x in val_instance]
    test_list = [(x.id, 0) for x in test_instance]

    splits.append((train_list, val_list))
    splits.append((train_list + val_list, test_list))

    return splits

Exemplo n.º 2

Arquivo: benchmark_db.py Projeto: lysh/temporal-segment-networks

def parse_activitynet_splits(version):
    db = ANetDB.get_db(version)
    train_instance = db.get_subset_instance('training')
    val_instance = db.get_subset_instance('validation')
    test_instance = db.get_subset_videos('testing')

    splits = []

    train_list = [(x.name, x.num_label) for x in train_instance]
    val_list = [(x.name, x.num_label) for x in val_instance]
    test_list = [(x.id, 0) for x in test_instance]

    splits.append((train_list, val_list))
    splits.append((train_list + val_list, test_list))

    return splits