Skip to content

Commit 3fb639e

Browse files
committed
Fix type hint in trim_and_resize_if_required function
1 parent 2c76b49 commit 3fb639e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

library/train_util.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2391,7 +2391,7 @@ def load_image(image_path, alpha=False):
23912391

23922392
# 画像を読み込む。戻り値はnumpy.ndarray,(original width, original height),(crop left, crop top, crop right, crop bottom)
23932393
def trim_and_resize_if_required(
2394-
random_crop: bool, image: Image.Image, reso, resized_size: Tuple[int, int]
2394+
random_crop: bool, image: np.ndarray, reso, resized_size: Tuple[int, int]
23952395
) -> Tuple[np.ndarray, Tuple[int, int], Tuple[int, int, int, int]]:
23962396
image_height, image_width = image.shape[0:2]
23972397
original_size = (image_width, image_height) # size before resize

0 commit comments

Comments
 (0)