VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update utils.py

Browse Source 
Cleanup
This commit is contained in:
Ed McManus
2024-09-19 14:38:32 -07:00
committed by GitHub
parent 3957372ded
commit de4bb657b0
1 changed files with 0 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
library/utils.py
View File

@@ -305,25 +305,19 @@ class MemoryEfficientSafeOpen:
raise ValueError(f"Unsupported float8 type: {dtype_str} (upgrade PyTorch to support float8 types)") raise ValueError(f"Unsupported float8 type: {dtype_str} (upgrade PyTorch to support float8 types)")
def pil_resize(image, size, interpolation=Image.LANCZOS): def pil_resize(image, size, interpolation=Image.LANCZOS):
# Check if the image has an alpha channel
has_alpha = image.shape[2] == 4 if len(image.shape) == 3 else False has_alpha = image.shape[2] == 4 if len(image.shape) == 3 else False
if has_alpha: if has_alpha:
# Convert BGRA to RGBA
pil_image = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGRA2RGBA)) pil_image = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGRA2RGBA))
else: else:
# Convert BGR to RGB
pil_image = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB)) pil_image = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
# Resize the image
resized_pil = pil_image.resize(size, interpolation) resized_pil = pil_image.resize(size, interpolation)
# Convert back to cv2 format # Convert back to cv2 format
if has_alpha: if has_alpha:
# Convert RGBA to BGRA
resized_cv2 = cv2.cvtColor(np.array(resized_pil), cv2.COLOR_RGBA2BGRA) resized_cv2 = cv2.cvtColor(np.array(resized_pil), cv2.COLOR_RGBA2BGRA)
else: else:
# Convert RGB to BGR
resized_cv2 = cv2.cvtColor(np.array(resized_pil), cv2.COLOR_RGB2BGR) resized_cv2 = cv2.cvtColor(np.array(resized_pil), cv2.COLOR_RGB2BGR)
return resized_cv2 return resized_cv2