Commit cb11e00c authored by Vít Starý Novotný's avatar Vít Starý Novotný
Browse files

Fail gracefully when input directory is empty

parent a527b721

ahisto_ocr/cli.py

+7 −4
Original line number Diff line number Diff line
@@ -63,6 +63,9 @@ def main(input_dir: str, output_dir: str, pero_ocr: bool, super_resolution: bool 
        in input_dir.glob('**/*.*')

    )



    if not input_images:

        LOGGER.info(f'No input images found in {input_dir}')

    else:

        output_dir = Path(output_dir)



        if gpus == 'auto':