def _get_slip_videos_and_annotations()

in pytouch/datasets/slip_sequence.py [0:0]

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

def _get_slip_videos_and_annotations(root, subset, prefix):
    with open(root, "r") as f:
        data_list = json.load(f)

    videos = []
    labels = []

    for obj in data_list:
        selected_subset = obj["subset"]
        if selected_subset == subset:
            if subset == "testing":
                # no label for testing data
                videos.append(os.path.join(prefix, obj["videopath"]))
            else:
                videos.append(os.path.join(prefix, obj["videopath"]))
                labels.append(obj["label"])
    return videos, labels