INNER CODE UNIT · Python
start_x
Shilin-LU/TF-ICON · scripts/app.py:59
start_x = (target_size[1] - segmentation_map_resize.shape[0]) // 2
start_y = (target_size[0] - segmentation_map_resize.shape[1]) // 2
padded_segmentation_map[start_x: start_x + segmentation_map_resize.shape[0], start_y: start_y + segmentation_map_resize.shape[1]] = segmentation_map_resize
# Create a new RGB image with the target size and place the resized image in the center
padded_image = Image.new("RGB", target_size)
start_x = (target_size[0] - image_resize.width) // 2
start_y = (target_size[1] - image_resize.height) // 2
padded_image.paste(image_resize, (start_x, start_y))
# Update the variable "image" to contain the final padded image
image = padded_image
else:
# image = Image.open(path).convert("RGB")
w, h = image.size
# print(f"loaded input image of size ({w}, {h}) from {path}")
w, h = map(lambda x: x - x % 64, (w, h)) # resize to integer multiple of 64
w = h = 512