dataset.py is now the one from plant-project
dataset.py: squashed indenting (hopefully no mistakes)