Fix GIF Image Handling in Dataset Loader

[bibisbar]

Fix GIF Image Handling in Dataset Loader

📝 Description

This PR improves the dataset image loading pipeline by adding support for GIF images and handling cases where the image file extension might be different than expected.

🔧 Changes

• Added automatic fallback to check for alternative image formats (.jpg, .jpeg, .png, .bmp, .webp, .gif) if the expected file is missing.
• Implemented proper handling for .gif images by converting the first frame to RGB mode to ensure compatibility.
• Wrapped the image loading process with a with Image.open(...) statement for better resource management.

🐞 Problem

Previously, the dataset loader assumed all images were in a fixed format (e.g., .jpg), leading to FileNotFoundError when images were stored with different extensions. Additionally, GIF files were not handled correctly, causing unexpected errors.

✅ Solution

• If the expected image file is missing, the code now iterates through possible extensions to locate the correct file.
• When loading a GIF file, it automatically converts the first frame to RGB, preventing format-related issues.

📌 Related Issues

• Fixes FileNotFoundError caused by missing or mismatched file extensions.
• Resolves compatibility issues with .gif images.

🔬 Testing

• Verified that images with different formats are correctly loaded.
• Confirmed that GIF images no longer cause errors and are properly converted.

Let me know if any further improvements are needed! 🚀