wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-30 00:00:26 +08:00
Code Issues Actions 20 Packages Projects Releases Wiki Activity

Fix transpose with non-square images

Browse Source
This commit is contained in:
missionfloyd 2023-04-09 19:19:14 -06:00 committed by GitHub
parent 97c7a402d7
commit d414157701
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy_extras/nodes_post_processing.py
View File

@ -232,6 +232,8 @@ class Transpose:
def transpose(self, image: torch.Tensor, method: str): def transpose(self, image: torch.Tensor, method: str):
batch_size, height, width, _ = image.shape batch_size, height, width, _ = image.shape
result = torch.zeros_like(image) result = torch.zeros_like(image)
if height != width and method in ("Transpose", "Transverse", "Rotate 90°", "Rotate 270°"):
result = torch.permute(result, (0, 2, 1, 3))
methods = { methods = {
"Flip horizontal": Image.Transpose.FLIP_LEFT_RIGHT, "Flip horizontal": Image.Transpose.FLIP_LEFT_RIGHT,