jgraving / DeepPoseKit

a toolkit for pose estimation using deep learning
http://deepposekit.org
Apache License 2.0
391 stars 90 forks source link

Potential draw_graph function fix for issue #52 #58

Open ghost opened 4 years ago

ghost commented 4 years ago

This is just a suggestion for a possible fix for the draw_graph function. The first thing it does is restrict edge_confidence to the indices where lines != -1 (that is, it restricts edge_confidence to the output layers corresponding to non-parent keypoints). I think this should work for all graphs - but maybe I've overlooked something. It at least works for the graph I described in issue #52. The second thing it does is check to see if the keypoints don't exist, in which case a line is not drawn.

stale[bot] commented 4 years ago

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs.