Support other types of input images

[femj007]

Currently, the code seems to only support *.png as input. And when I use other types as input, I get an error like "IndexError: list index out of range".

Here I provide a simple idea to support other types of input images like *.jpg or *.bmp. It works fine on my machine.