fix: 🐛 tensor2pil

2023-07-21 23:37:12 +02:00
parent aa551ebe57
commit 8a59508ff9
1 changed files with 2 additions and 1 deletions
@@ -53,7 +53,8 @@ def tensor2pil(image: torch.Tensor) -> List[Image.Image]:

    if batch_count > 1:
        out = []
-        out.extend([tensor2pil(image[i]) for i in range(batch_count)])
+        for i in range(batch_count):
+            out.extend(tensor2pil(image[i]))
        return out

    return [