torch_concepts.data.datasets.traffic.TrafficLights

class TrafficLights(root_dir='./data_cache/', split='train', concept_transform=None, class_dtype=<class 'float'>, img_transform=None, regenerate=False, sym_link=None, use_absolute_path=False, num_threads=4, n_samples=1000, seed=42, position_para_noise=50, position_perp_noise=20, error_probability=0.1, p_ambulance=0.2, min_num_cars=0, max_num_cars=7, resize_final_image=0.15, car_colors=['black', 'blue', 'burgundy', 'green', 'pink', 'purple', 'silver', 'white'], possible_starting_directions=['north', 'east', 'south', 'west'], thickness=100, light_scale=1.5, use_lights_sprites=False, train_ratio=0.6, val_ratio=0.2, test_ratio=0.2, test_config_override_values=None, val_config_override_values=None, selected_concepts=None)

Synthetic traffic dataset

__init__(root_dir='./data_cache/', split='train', concept_transform=None, class_dtype=<class 'float'>, img_transform=None, regenerate=False, sym_link=None, use_absolute_path=False, num_threads=4, n_samples=1000, seed=42, position_para_noise=50, position_perp_noise=20, error_probability=0.1, p_ambulance=0.2, min_num_cars=0, max_num_cars=7, resize_final_image=0.15, car_colors=['black', 'blue', 'burgundy', 'green', 'pink', 'purple', 'silver', 'white'], possible_starting_directions=['north', 'east', 'south', 'west'], thickness=100, light_scale=1.5, use_lights_sprites=False, train_ratio=0.6, val_ratio=0.2, test_ratio=0.2, test_config_override_values=None, val_config_override_values=None, selected_concepts=None)

Methods

__init__([root_dir, split, ...])
sample_array(real_idx)