Imported from https://github.com/rickyHong/pytorch-light-head-rcnn-repl