gather_data correction

[Kkoonnrr]

After you remove one of the train/test/validation data or pickle folders and keep the rest, gather_data.py won't work properly.