Fixed a bug that caused images to be separated

2024-05-29 14:21:37 +03:30
parent e8e11c384e
commit 2c2f182653
1 changed files with 3 additions and 1 deletions
@@ -747,4 +747,6 @@ class pytorch_openpose:
            output_image, _ = process_one (input_image)  
            img = pil2tensor(output_image) 
            results.append(img)
-        return (torch.cat(results, dim=0),)
+            # return img
+        return (torch.cat(results, dim=0),)
+