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

Merge branch 'dev' into sd3

Browse Source
This commit is contained in:
kohya-ss
2024-08-13 21:00:44 +09:00
parent 4cf42cc5d4 74f91c2ff7
commit f5ce754bc2
3 changed files with 16 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
library/train_util.py
View File

@@ -2638,16 +2638,20 @@ def load_arbitrary_dataset(args, tokenizer) -> MinimalDataset:
return train_dataset_group
def load_image(image_path, alpha=False):
image = Image.open(image_path)
if alpha:
if not image.mode == "RGBA":
image = image.convert("RGBA")
else:
if not image.mode == "RGB":
image = image.convert("RGB")
img = np.array(image, np.uint8)
return img
def load_image(image_path, alpha=False):
try:
with Image.open(image_path) as image:
if alpha:
if not image.mode == "RGBA":
image = image.convert("RGBA")
else:
if not image.mode == "RGB":
image = image.convert("RGB")
img = np.array(image, np.uint8)
return img
except (IOError, OSError) as e:
logger.error(f"Error loading file: {image_path}")
raise e
# 画像を読み込む。戻り値はnumpy.ndarray,(original width, original height),(crop left, crop top, crop right, crop bottom)