From 2c2f182653c4fc097f9c8d9b86ee0951d4af1d01 Mon Sep 17 00:00:00 2001 From: nirex0 Date: Wed, 29 May 2024 14:21:37 +0330 Subject: [PATCH] Fixed a bug that caused images to be separated --- pytorch_openpose.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pytorch_openpose.py b/pytorch_openpose.py index c894e6e..6947c19 100644 --- a/pytorch_openpose.py +++ b/pytorch_openpose.py @@ -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),) \ No newline at end of file + # return img + return (torch.cat(results, dim=0),) + \ No newline at end of file