hourglasstensorflow
hourglasstensorflow copied to clipboard
Added format and scale
Added a function to format and re-scale the predictions.
formatAndScale.py does 2 things:
-
Changes from List to Dict, so you can access the coordinates using the name of a body part, i.e. "Right Ankle"
-
Scales the data from 256x256 back to your original height and width For example, if original image is 1280x720, the returned result will be the X,Y coordinates for a 1280x720 image, rather than for a 256x256 image
It takes the prediction from hourglass, plus the original height and width, as input. Returns a dict of coordinates, scaled back to original size.