color/visualize_dataset.py