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