Imported from https://github.com/dreamway/CenterNet-objects-as-points