pytorch implementation of openpose including Hand and Body Pose Estimation.
2.07k
stars
395
forks
source link
TypeError: can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first. #16
Closed
12345k closed 4 years ago
I fixed this by changing the line in python/hand.py: output = self.model(data).numpy()
to: output = self.model(data).cpu().numpy()