Training code for "Pixels to Graphs by Associative Embedding" - View it on GitHub