Добавить разбивку на train/valid/test.

Код разметки возвращает полный датасет, код сегментации принимает train/test/valid - нужна возможность связать в единый пайпалайн внутри библиотеки. 

Возможно, в случае если нужно решить одновременно задачи и сегментации и классификации (т.е. изначальные изображения находятся в нескольких папках), то добавить переименование изображений, включая название папки в имя файла. 