Why don't you try using the pathlib library?

[wakame1367]

I think we can swap out img_path.split('.')[-1].lower() for [PurePath.suffix].(https://docs.python.org/3/library/pathlib.html#pathlib.PurePath.suffix)

https://github.com/Soongja/basic-image-eda/blob/77cc1b3d2849776647acda2eea8241dddbaf4bba/basic_image_eda/eda.py#L50

reference_link

• Python - pathlib