一个用于将任何扩展图像作为输入的线性 Python 代码

如何解决一个用于将任何扩展图像作为输入的线性 Python 代码

在下面的代码中，我正在考虑只输入 .jpg 文件。但我想重新格式化代码，以便输入任何扩展名或多个扩展名，如 .png、.JPEG 等硬编码。 虽然我更愿意允许像我上面提到的那样的特定扩展以及 2-3 个更多。

inp_img = cv2.imread(f'input_images/{IMAGE_NAME}_{i}.jpg',cv2.IMREAD_UNCHANGED)

感谢您的考虑。