- Replace NaN and Inf values with valid values before saving
- Clip image values to valid range [0, 1] to prevent black images