Filtering bdd100k dataset by timeofday attribute

bdd100k_labels_filter_of_timeofday.py

import json

def filter_train():
    data = []
    with open('bdd100k_labels/bdd100k_labels_images_train.json') as f:
        train = json.load(f)
        for image in train:
            if image['attributes']['timeofday'] == 'daytime':
                data.append(image)
            
    new_file = open('bdd100k_labels/bdd100k_labels_images_train_daytime.json', 'w')
    json.dump(data, new_file, indent=4)
    new_file.close()

def filter_val():
    data = []
    with open('bdd100k_labels/bdd100k_labels_images_val.json') as f:
        train = json.load(f)
        for image in train:
            if image['attributes']['timeofday'] == 'daytime':
                data.append(image)
            
    new_file = open('bdd100k_labels/bdd100k_labels_images_val_daytime.json', 'w')
    json.dump(data, new_file, indent=4)
    new_file.close()

if __name__ == '__main__':
    filter_train()
    filter_val()