Добавить в функции обработки проверку на тип файла

В segment_images_from_folder если попадается в папке не изображение (обычно временные автоматически создаваемые файлы), то вылетает ошибка.