Configure image extensions

[kevinsuedmersen]

Hi,

Thanks for this great repo, it works really great. I would suggest to make the image extensions in the get_img_paths function configurable. Otherwise you will get an IndexError later on in the code (in line 378 I believe), if your images happen to have another file extension, like .bmp for example