Imported from https://github.com/huanghoujing/AlignedReID-Re-Production-Pytorch