search

Hzzone / pytorch-openpose

pytorch implementation of openpose including Hand and Body Pose Estimation.
2.07k stars 395 forks source link

how to generate the graph like this? #78

Closed shu0o0 closed 1 year ago

shu0o0 commented 1 year ago

@Hzzone Hello sir, the figure on the left is obtained body.py run, I wonder if this code can generate the graph on the right, if so, can you tell me the method?

image

Hzzone commented 1 year ago

See https://github.com/Hzzone/pytorch-openpose/blob/5ee71dc10020403dc3def2bb68f9b77c40337ae2/src/body.py#L216 change to

 canvas = util.draw_bodypose(np.zeros_like(oriImg), candidate, subset)
shu0o0 commented 1 year ago

See

https://github.com/Hzzone/pytorch-openpose/blob/5ee71dc10020403dc3def2bb68f9b77c40337ae2/src/body.py#L216

change to

 canvas = util.draw_bodypose(np.zeros_like(oriImg), candidate, subset)

OK, I get it, Thank you very much, it will help me much!